[Install] RTX 3090, cuda 11.2 Pytorch 설치

※ Docker container 안에 cuda와 호환되는 pytorch 설치 

※ [Error] no kernel image is available for execution on the device CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect. 에러 해결법

 

※ 사용환경 : RTX 3090, cuda 11.2

 

- 본인의 pip 환경에 따라서 pip, pip3 선택해서 설치

pip : python2 

pip3 : python3 

Error: Command 'pip3' not found, but can be installed with 에러가 뜨는 경우

$ apt-get install -y python3-pip

 

python3도 없다면

$ apt-get install -y python3 하고, 그 다음에 pip 설치하기

 

$ pip2 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

 

$ pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

 

- 설치 확인