정보성) cuda11.4 와 맞는 torch version 2.0.1

 

cuda 11.4일 때 pytorch 2.0 대를 쓰고 싶어서 인터넷에서 찾아봤고 공유드린다.

 

바로 2.0.1이다(2024.02.25 기준) (python 3.10)

pip install torch==2.0.1

 

 

그렇다면 어떻게 찾을 수 있을까?

- 모르겠다..ㅎ

 

일단 재 설치되는 기준으로 보면 아래와 같은 것들이 설치가 되기는 했다. 

pip list | grep cuda

cuda 11.7 관련한 것들을 자동으로 설치하는 것 같다.

Collecting nvidia-cuda-nvrtc-cu11==11.7.99 (from torch==2.0.1)
  Downloading nvidia_cuda_nvrtc_cu11-11.7.99-2-py3-none-manylinux1_x86_64.whl.metadata (1.5 kB)
Collecting nvidia-cuda-runtime-cu11==11.7.99 (from torch==2.0.1)
  Downloading nvidia_cuda_runtime_cu11-11.7.99-py3-none-manylinux1_x86_64.whl.metadata (1.6 kB)
Collecting nvidia-cuda-cupti-cu11==11.7.101 (from torch==2.0.1)
  Downloading nvidia_cuda_cupti_cu11-11.7.101-py3-none-manylinux1_x86_64.whl.metadata (1.6 kB)
Collecting nvidia-cudnn-cu11==8.5.0.96 (from torch==2.0.1)
  Downloading nvidia_cudnn_cu11-8.5.0.96-2-py3-none-manylinux1_x86_64.whl.metadata (1.6 kB)
Collecting nvidia-cublas-cu11==11.10.3.66 (from torch==2.0.1)
  Downloading nvidia_cublas_cu11-11.10.3.66-py3-none-manylinux1_x86_64.whl.metadata (1.6 kB)
Requirement already satisfied: nvidia-cufft-cu11==10.9.0.58 in /opt/conda/envs/llm/lib/python3.10/site-packages (from torch==2.0.1) (10.9.0.58)
Collecting nvidia-curand-cu11==10.2.10.91 (from torch==2.0.1)
  Downloading nvidia_curand_cu11-10.2.10.91-py3-none-manylinux1_x86_64.whl.metadata (1.6 kB)
Collecting nvidia-cusolver-cu11==11.4.0.1 (from torch==2.0.1)
  Downloading nvidia_cusolver_cu11-11.4.0.1-2-py3-none-manylinux1_x86_64.whl.metadata (1.6 kB)
Collecting nvidia-cusparse-cu11==11.7.4.91 (from torch==2.0.1)
  Downloading nvidia_cusparse_cu11-11.7.4.91-py3-none-manylinux1_x86_64.whl.metadata (1.6 kB)
Collecting nvidia-nccl-cu11==2.14.3 (from torch==2.0.1)
  Downloading nvidia_nccl_cu11-2.14.3-py3-none-manylinux1_x86_64.whl.metadata (1.8 kB)
Collecting nvidia-nvtx-cu11==11.7.91 (from torch==2.0.1)
  Downloading nvidia_nvtx_cu11-11.7.91-py3-none-manylinux1_x86_64.whl.metadata (1.7 kB)
Collecting triton==2.0.0 (from torch==2.0.1)
  Downloading triton-2.0.0-1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl.metadata (1.0 kB)

 

실제 설치된 버전은 다음과 같다.

print(torch.version.cuda)
print(torch.__version__)
print(torch.cuda.is_available())

# 11.7
# 2.0.1+cu117
# True

 

 

 

 

 

 

다른 분들도 참고하시면 좋을 것 같아 글을 남깁니다 :)

 

 

https://download.pytorch.org/whl/torch/

https://download.pytorch.org/whl/torch/

https://discuss.pytorch.org/t/which-pytorch-version-2-0-1-support-cuda-11-4/190446/3

Which pytorch version >2.0.1 support cuda 11.4