VSCode에서 Cuda를 사용하려고 할 때 아래와 같은 경고가 출력된다면
Cuda버전과 맞는 pytorch를 설치 해야합니다.
Cuda 버전은 nvidia-smi
을 치면 확인 가능합니다.
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia
pytorch를 설치하려고 했는데 아래와 같은 경고문이 떴다면
conda update -n base -c defaults conda
이렇게 Conda업데이트를 해야하는데 업데이트가 안될 수 있습니다.
이럴 때는 anaconda Prompt에서 conda update -n base -c defaults conda
를 실행하면 conda 업데이트가 진행됩니다.
업데이트가 끝났다면 다시 VSCode로 돌아와 pytorch를 설치하면 됩니다.