반응형
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
'Deep Learning' 카테고리의 다른 글
Miniconda로 Deep learning 가상환경 만들기 (0) | 2023.03.22 |
---|---|
Anaconda의 유료화에 따른 Miniconda+Conda_forge로 이동 (무료화) (0) | 2023.03.20 |
Anaconda를 이용한 env 생성 및 제거 (0) | 2023.03.08 |
Gaussian, LogGaussian PDF in Pytorch (0) | 2022.12.06 |
Implement KL Divergence using Pytorch (0) | 2022.12.06 |