CUDA 11.2 + RTX3090에 PyTorch 설치

(2021-08-27) 기준

Environment

• Ubuntu 20.04
• RTX 3090

Version

• CUDA: 11.2 / cuDNN: 8.1.1
• Python: 3.8.10
• PyTorch: 1.7.1 / Torchvision: 0.8.2

Installation

Problem

지금까지 나온 버전에서 11.2로 나온 버전이 없었기 때문에, 11.x를 설치하면 되지 않을까? 하고 설치를 하고자 했습니다. 저는 torch는 보통 conda 명령어로 설치를 하기 때문에 사이트에 안내된 명령어로 설치를 했더니,

# CUDA 11.0
conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=11.0 -c pytorch

import torch를 하자마자,

OSError: /home/psl/.conda/envs/imb/lib/python3.8/site-packages/torch/lib/../../../../libcublas.so.11: undefined symbol: free_gemm_select, version libcublasLt.so.11 와 같은 에러가 발생했습니다.

Solution

인터넷을 검색하다보니, https://powerofsummary.tistory.com/189 에서 wheel 부분에 있는 명령어인

# CUDA 11.0
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html

• https://pytorch.org/get-started/previous-versions/

을 설치해서 실행이 된다하여 진행하고 있던 conda env를 삭제후, 새로 만들어 새로운 환경에서 위의 명령어로 설치를 했습니다.

실제로, 저렇게 입력을 하니까 import torch가 문제 없이 실행되었고… 어떠한 이유에서 conda는 안되고 pip로는 실행이 되는지 잘 모르겠습니다…

Reference

• https://powerofsummary.tistory.com/189