win10 + Python3.6 + cuda8.0 + cuDNN6.0 + pytorch1.01/tensorflow-gpu安裝

win10 + Python3.6 + cuda8.0 + cuDNN6.0 + pytorch1.01/tensorflow-gpu安裝

1 Anaconda的安裝

官網下載安裝windows版本(64Bit)官網只有最新版Python3.7)
在這裏插入圖片描述
或者在https://repo.continuum.io/archive/下載各種版本的Annaconda,其中:Anaconda3-5.2.0-Windows-x86_64.exe 對應 python3.6.5,選擇下載安裝。

2 CUDA 、cuDNN的安裝

首先查看自己電腦的顯卡類型(我的是GTX 960M,安裝CUDA9.0沒成功,就換成8.0了),
然後,查看顯卡版本信息:https://developer.nvidia.com/cuda-gpus 在這裏插入圖片描述
在這裏插入圖片描述
(1)CUDA:https://developer.nvidia.com/cuda-toolkit-archive 選擇CUDA Toolkit8.0 - Feb 2017,window-local版本
在這裏插入圖片描述
(2)CuDNN:https://developer.nvidia.com/cudnn(需要註冊),如果安裝tensorflow-gpu的版本低於1.2.1用得到是cuDNN5.1,更高的tensorflow版本需要cuDNN 6.0。
在這裏插入圖片描述
(3)tensorflow
我沒用用國內鏡像源,網好的話可以直接下載,我也沒有在envs裏配置一個tensorflow的環境,直接利用 Anaconda Prompt 或者 cmd 進行安裝,這樣用起來比較方便:

conda install tensorflow-gpu ==1.3.0

或者

pip install tensorflow-gpu ==1.3.0

直接 conda install tensorflow-gpu 下載的是2.00版本,之前用的cpu版本是這個,好多函數都改了,用起來不是很方便,所以需要指定下載版本。對應關係如下圖(不是我的圖,原文:https://blog.csdn.net/sunmingyang1987/article/details/102872658)
這不是我的圖
進行測試:

import tensorflow as tf
sess = tf.Session()
a = tf.constant(10)
b= tf.constant(12)
sess.run(a+b)

在這裏插入圖片描述

以上參考:https://blog.csdn.net/qq_34438969/article/details/88430901

Pytorch 安裝

在Anaconda Prompt 中輸入:

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

安裝速度比較慢,我的 pytorch 下載了兩次才成功

安裝成功:
在這裏插入圖片描述
測試:
在這裏插入圖片描述

參考:
[1] https://blog.csdn.net/qq_34438969/article/details/88430901
[2] https://blog.csdn.net/sunmingyang1987/article/details/102872658
[3] https://blog.csdn.net/qq_36165459/article/details/88368573

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章