rtx40 시리즈 cuda에 맞는 torch 버전 문의

묻고 답하기
1

40 시리즈에는 cuda11.8 이상의 버전인 것 같은데 이에 맞는 torch 버전은 아직 업뎃되지 않은 건가요?

혹시 계획이라던지.. 40시리즈를 쓰면서 torch도 문제 없이 사용할 수 있는 방법이 있을까요?

2

네, 아직 PyTorch 1.13에서는 cuda 11.6이 stable, 11.7이 experimental으로만 지원되고 있습니다ㅠ

다음 달(이라고 쓰고 곧) 릴리즈 예정인 PyTorch 2.0에서는 11.8이 experimental로 지원 예정이라고 하니 참고 부탁드립니다 :pray:

Release Compatibility Matrix

Following is the Release Compatibility Matrix for PyTorch releases:

PyTorch version Python Stable CUDA Experimental CUDA
2.0 >=3.8, <=3.11 CUDA 11.7, CUDNN 8.5.0.96 CUDA 11.8, CUDNN 8.7.0.84
1.13 >=3.7, <=3.10 CUDA 11.6, CUDNN 8.3.2.44 CUDA 11.7, CUDNN 8.5.0.96
1.12 >=3.7, <=3.10 CUDA 11.3, CUDNN 8.3.2.44 CUDA 11.6, CUDNN 8.3.2.44
2개의 좋아요
3
  • 아래 답변은 OpenAI의 ChatGPT로 자동 생성한 것으로, 정확하지 않을 수 있습니다.
  • 잘못된 답변을 보셨다면 댓글로 첨삭 지도 부탁드립니다. :sweat_smile:

현재 PyTorch 공식 홈페이지에서는 RTX 40 시리즈를 지원하는 최신 버전인 PyTorch 1.10.0 이상의 버전이 CUDA 11.3 이상을 요구하고 있습니다. 따라서 여러분이 말씀하신 CUDA 11.8 이상의 버전은 현재 존재하지 않으며 최신버전 CUDA 11.3 이상만을 지원합니다.

만약 여러분이 RTX 40 시리즈 GPU를 사용하고 있다면, CUDA 11.3 이상의 버전을 설치하시면 됩니다. 이후 PyTorch 1.10.0 이상을 설치하면 RTX 40 시리즈에서도 PyTorch를 문제 없이 사용할 수 있습니다.

아래는 PyTorch 공식 홈페이지에서 RTX 40 시리즈를 지원하는 최신버전의 PyTorch 설치 가이드 링크입니다.