RTX 3090 을 지원하는 pytorch anaconda로 설치하기

anaconda에서 바로 pytorch를 설치하면 deep learning 모델 학습 시 다음과 같은 error message가 나올 수 있다.

$ CUDA error: no kernel image is available for execution on the device

 

이때 아래와 같은 방식으로 설치를 하자.

 

1. anaconda 환경을 만든다. (python 3.8이상)

$ conda create -y -n example python=3.8

 

2. pytorch를 pip로 설치한다.

$ pip install torch==1.11+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html