반응형
CUDA 버전별 설치가능한 Tensorflow, PyTorch 버전 확인하는 방법
Deep Learning & Machine Learning/Tensorflow&PyTorch 개발환경2024. 3. 14. 21:32CUDA 버전별 설치가능한 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 설치하..

반응형
image