pytorch cuda 설치

cuda toolkit과 pytorch version이 맞지 않은 것이다. (서로 호환이 되지 않는다는 말이다.)

 

cuda랑 일주일동안 싸웠다.

진짜 별게 아닌데... 좀 오래 싸웠다.

 

일주일동안 안됐던 이유 (예상) :

1. 설치 순서를 지키지 않았다.

    - cuda toolkit을 먼저 깔고 cudnn을 깔아주어야 cudnn의 버전이 자동으로 지정되어 설치되는 느낌?

2. 패키지 전체를 다운로드 받은 것이 아닌, 모듈 하나 하나 다운 받아서 버전이 서로 호환되지 않았던 것 같다.

3. toolkit 설치할 때 버전을 지정해주지 않고 다운로드한 것이 문제의 시작이었던 것 같다.

 

순서: cuda toolkit - cudnn - pytorch

• cuda toolkit

conda install -c "nvidia/label/cuda-11.7.0" cuda-toolkit

 

• cudnn

conda install -c anaconda cudnn

 

• pytorch

// nvidia-smi에 나오는 추천 버전으로 설치한다.

여기로 들어가서 

버전에 맞는 torch를 설치해준다. pip install 통해서.

꼭꼭 

+cu117 있어야 한다. (그래야 GPU 활용 가능함.)

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

https://pytorch.org/

PyTorch