Pytorch_gpu_설치

stackoverflow에 질문 한 것 (cudatoolkit 설치 과정)
교정된 질문


설치한 cuda 버전

pytorch=1.1.0

-> python [version= '>=3.5, <3.6.0a0 | >=3.6,<3.7.0a0 | >=3.7,<3.8.0a0' ]

 

torchvision=0.3.0

-> python[version= '>=3.5,<3.6.0a0 | >=3.6,<3.7.0a0 | >=3.7,<3.8.0a0' ]

 

<제어판에 있는 cuda 관련 프로그램>

파이썬 버전을 3.8 --> 3.7.7로 다운그레이드, 아나콘다 터미날에서

conda install python=3.7 # Stackoverflow에 질문한 것에 대해 해결

 

다시

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

 

성공!!

 

tensorflow-gpu 설치

conda install -c anaconda tensorflow-gpu

작동되지 않아서 다시 아래 설치, ‘tf-gpu-cuda8’이 가상환경인지 모르고 설치

conda create -n tf-gpu-cuda8 tensorflow-gpu cudatoolkit=9.0

conda activate tf-gpu-cuda8

텐서플로우가 작동은 하는데 한묶음의 에러 발생, 다시

pip install "numpy<1.17" --> 설치, 그러나 에러메시지, 다시

conda install -c anaconda google-pasta

 

모든 것 ok!!