When will you grow up?

Pytorch 1.1.0 / Tensorflow-gpu 1.12 설치 및 cuda10_9 확인 본문

02. Study

Pytorch 1.1.0 / Tensorflow-gpu 1.12 설치 및 cuda10_9 확인

미카이 2020. 1. 9. 13:38

Pytorch 1.1.0 version

# CUDA 9.0

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

# CUDA 10.0

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

# CPU Only

conda install pytorch-cpu==1.1.0 torchvision-cpu==0.3.0 cpuonly -c pytorch

 

 

 

Tensorflow-gpu 1.12.0 version

conda install tensorflow-gpu=1.12

 

 

위 명령어로 설치할 수 있으며,

cuda 9.0버전을 사용하며,

python version은 3.6.9를 사용한다.

 

cuda 9.0cuda_9.0.176_win10 을 다운받았으며,

cudnn은 cudnn-9.0-windows10-x64-v7.3.1.20 이 가장 잘 어울리고 오류없이 작동하는것을 확인하였다.

https://developer.nvidia.com/cuda-toolkit-archive -> cuda 설치

https://developer.nvidia.com/rdp/cudnn-archive -> cudnn 설치

 

 

이렇게 설치하니 gpu로 잘 동작되는것을 확인하였다.

gpu 동작확인은 클릭

 

--------------------------------------------------------------------------------------------------------------------

번외로

요즘 Tensorflow 2.0 버전을 많이 사용하는데,

내가 사용하는 cuda10은 아래와 같다

 

Tensorflow 2.0 설치

pip install tensorflow-gpu==2.0

 

위 명령어로 설치할 수 있으며,

cuda 10.0버전을 사용하며,

python version은 3.6.9를 사용한다.

 

cuda_10.0.130_411.31_win10 을 다운받았으며,

cudnn은 cudnn-10.0-windows10-x64-v7.6.0.64 이 가장 잘 어울리고 오류없이 작동하는것을 확인하였다.

https://developer.nvidia.com/cuda-toolkit-archive -> cuda 설치

https://developer.nvidia.com/rdp/cudnn-archive -> cudnn 설치

 

cuda10.1 보단 10.0을 안전하게 설치하자 ~_~

 

시간 여유가 되면, cuda 10/9 설치해서 동시에 사용은 안되지만 path 잡아서 변경하는것을 포스팅하려한다...

이것때매 삽질을 많이했다 ㅠㅠ...

Comments