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)
win10 + Python3.6 + cuda8.0 + cuDNN6.0 + pytorch1.01/tensorflow-gpu安装_第1张图片
或者在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 win10 + Python3.6 + cuda8.0 + cuDNN6.0 + pytorch1.01/tensorflow-gpu安装_第2张图片
win10 + Python3.6 + cuda8.0 + cuDNN6.0 + pytorch1.01/tensorflow-gpu安装_第3张图片
(1)CUDA:https://developer.nvidia.com/cuda-toolkit-archive 选择CUDA Toolkit8.0 - Feb 2017,window-local版本
win10 + Python3.6 + cuda8.0 + cuDNN6.0 + pytorch1.01/tensorflow-gpu安装_第4张图片
(2)CuDNN:https://developer.nvidia.com/cudnn(需要注册),如果安装tensorflow-gpu的版本低于1.2.1用得到是cuDNN5.1,更高的tensorflow版本需要cuDNN 6.0。
win10 + Python3.6 + cuda8.0 + cuDNN6.0 + pytorch1.01/tensorflow-gpu安装_第5张图片
(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)
win10 + Python3.6 + cuda8.0 + cuDNN6.0 + pytorch1.01/tensorflow-gpu安装_第6张图片
进行测试:

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 下载了两次才成功

安装成功:
win10 + Python3.6 + cuda8.0 + cuDNN6.0 + pytorch1.01/tensorflow-gpu安装_第7张图片
测试:
在这里插入图片描述

参考:
[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

你可能感兴趣的:(Python,tensorflow,cuda,anaconda,深度学习,pytorch)