CUDA 버전별 설치가능한 Tensorflow, PyTorch 버전 확인하는 방법

CUDA 버전별로 설치가능한 Tensorflow, PyTorch 버전을 확인하는 방법을 다룹니다. 

2024. 3. 14 최초작성

 

 

 

Tensorflow

 

CUDA 버전별로 설치가능한 Tensorflow 버전은 아래 링크에서 확인할 수 있습니다.  

https://www.tensorflow.org/install/source?hl=ko#gpu  

 

가급적이면 Tensorflow 버전에 맞는 CUDA를 설치하는 것을 권장하는 것으로 보입니다.  Tensorflow와 CUDA를 같이 설치해주는 tensorflow[and-cuda] 패키지를 설치해보면 Tensorflow 버전에 따라 설치되는 CUDA 버전이 다릅니다.  CUDA 버전이 적절하지 않으면 Tensorflow에서 GPU를 사용할 수 없습니다. 

Tensorflow 설치하는 방법은 다음 포스트를 참고해보세요.

 

Ubuntu 22.04에 CUDA 사용하도록 Tensorflow 설치하는 방법

https://webnautes.tistory.com/1878

 

윈도우에서 Tensorflow 2.11 이상 버전을 설치하려면 WSL2 설치가 필요합니다. 

 

WSL2에 CUDA 사용하는 Tensorflow 설치하는 방법

https://webnautes.tistory.com/1873 

PyTorch

 

PyTorch의 경우엔 아래 링크에서 최신 버전의 PyTorch와 호환성 있는 CUDA 버전을 선택하면 설치 방법을 알려줍니다. PyTorch와 CUDA가 같이 설치됩니다. 

https://pytorch.org/get-started/locally/

 

글 작성 시점(2024. 3. 14)에 PyTorch 2.2.1을 설치할 수 있는 CUDA 버전은 CUDA 11.8과 CUDA 12.1입니다.  CUDA 11.8을 선택하느냐 CUDA 12.1을 선택하느냐에 따라 설치방법을 그에 맞게 알려줍니다.

 

이전 버전들의 설치방법은 아래 링크에 모아져 있습니다. 

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

PyTorch 버전별로 (아래 스크린샷은 2.2.0입니다.)  conda를 위한 명령이 있는 Conda 항목과 pip를 위한 명령이 있는 Wheel 항목이 있습니다. Conda 항목에 있는 것은 Miniconda/Anaconda 환경에서 설치시 사용할 수 있는 방법입니다. 

 

Pytorch 설치하는 방법은 다음 포스트를 참고해보세요.

 

Ubuntu 22.04에 CUDA 사용할 수 있도록 PyTorch 설치하는 방법

https://webnautes.tistory.com/1845

 

Windows에 CUDA 사용할 수 있도록 PyTorch 설치하는 방법

https://webnautes.tistory.com/1850