728x90
반응형
일단 기본적으로 파이토치를 설치하고 잘 연결되어있는지 확인하기 위해 해당 코드를 실행한다
import torch
print("Torch version:{}".format(torch.__version__))
print("cuda version: {}".format(torch.version.cuda))
print("cudnn version:{}".format(torch.backends.cudnn.version()))
print(torch.cuda.is_available())
나는 처음에 해당 코드를 실행시 쿠다와 cuDNN 을 설치했음에도 안잡혔었다..
-우분투 18.04 기준으로
CUDA 버전 확인
nvcc -V
cuDNN 버전 확인
cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2
CUDA와 cuDNN 이 정상적으로 설치 되어 있는 경우라면 파이토치 버전이 잘못된것임..
파이토치 버전 뒤에 +cu116 과 같이 쿠다 버전이 적혀야한다
그래서 다운그레이드 해줌
해결방법
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
https://pytorch.org/get-started/previous-versions/
PyTorch
An open source machine learning framework that accelerates the path from research prototyping to production deployment.
pytorch.org
해당사이트에서 자신의 쿠다 버전과 OS 환경에 맞게 설치해주면 된다.
최종적으로 이렇게 정상적이게 출력되면 끝!!
728x90
반응형
'OS > Ubuntu' 카테고리의 다른 글
[anaconda] 다른 계정의 아나콘다 사용하기 (0) | 2023.03.30 |
---|---|
[Linux] ssh 연결이 안될때 / 서버 스토리지 변경 후 아이피 연결 실패시 해결법 (0) | 2023.03.29 |
[Linux] 일반 계정에 관리자 권한 주기 (sudo 권한) (0) | 2023.03.29 |
[Linux] Obuntu 18.04 Root 비밀번호 분실시 변경하는 법 (0) | 2023.03.29 |
[Linux] 리눅스 백업 & 복구 (0) | 2023.03.27 |