CUDA 11.4에서 pytorch 설치하기

새 가상환경을 만든 후 pytorch를 설치하려는데 계속되는 버전 문제.

공식 홈페이지에서 cuda 11.3 버전으로 설치하니 되었다

 

pytorch 공식 홈페이지에서 previous versions을 참고하였다

https://pytorch.org/get-started/previous-versions/

Previous PyTorch Versions

아래에 나와있는 cuda 11.3 버전으로 설치하면 된다

# CUDA 11.3
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

pytorch가 정상적으로 설치가 되었는지 import와 간단한 연산으로 확인해본다

~$ python -c "import torch; print(torch.__version__)"
1.12.1
~$ python -c "import torch; print(torch.cuda.is_available())"
True
~$ python
>>> import torch
>>> x = torch.rand(5, 3).cuda()
>>> y = x * x + 2
>>> print(y)
tensor([[2.7343, 2.0482, 2.2066],
        [2.2666, 2.7418, 2.0105],
        [2.3232, 2.1034, 2.0014],
        [2.3596, 2.1746, 2.2879],
        [2.0381, 2.0335, 2.6469]], device='cuda:0')
>>>