WSL2 conda환경에서 cuda 설치하기

원래 pytorch gpu버전을 설치할때 tensorflow-gpu를 설치하고, 설치된 cudatookit에 맞는 pytorch를 설치했는데,

wsl환경에서 계속 cpu버전의 torch가 설치되는 문제가 발생했다.

https://discuss.pytorch.org/t/trouble-installing-torch-with-cuda-using-conda-on-wsl2/136911/4

위 게시물에서 pip을 이용해 설치하니 해결됐다고 하여

1. tensorflow-gpu 대신

conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn

위를 이용해 따로 cudnn, cudatoolkit을 설치 후

2. 설치된 cudatoolkit 버전에 맞는 torch를 pip을 이용해 설치했더니 해결되었다..!
   torch get-started

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

끼얏호~~ 👻👻👻