* CUDA11.2 버전에 적합한 pytorch를 설치해야 GPU 사용 가능
- https://pytorch.org/get-started/locally/
- 접속해서 아래와 같이 선택
- 현재 서버의 CUDA 버전은 (nvcc -V 로 확인) 11.2 이지만 11.2에 적합한 pytorch는 몇몇 블로그를 확인해보니 아래와 같이 설치하라고 되어있음pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html
- 하지만 위와 같이 해보니 pytorch를 불러오지 못함
- 따라서 아래와 같이 CUDA 11.6 버전용 pytorch를 설치하고 기도메타 수행
1 | conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia |
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html