Win10+Anaconda+CUDA+cuDNN安装CPU和GPU版本的TensorFlow

1.安装Anaconda

到官网直接下载:Anaconda下载

建议下载3.6版本,支持Python3.6

至于安装就没有特别需要注意的了,主要两点建议吧:

1)不要自己单独安装Python,建议在Anaconda的时候,它提示安装Python的时候打勾就行了,这样就会很好的相互匹配

2)勾上环境变量

è¿éåå¾çæè¿°

 

2.安装CUDA

安装CUDA也没有什么特别注意的,建议先更新NVIDIA显卡驱动

 

3.安装cuDNN

将解压过后的三个文件拷贝到CUDA的安装路径下即可

 

4.安装CPU版本TensorFlow

1)打开Anaconda Navigator

2)点击Environments

3)点击Create

4)给环境取名--tensorflow(随意)

5)在自己电脑的C盘用户目录下新建文件夹pip,里面新建文档pip.ini

6)pip.ini输入:

[global]
trusted-host=mirrors.aliyun.com
index-url=http://mirrors.aliyun.com/pypi/simple/

5和6步的目的在于更换pip的下载镜像,可以大大提高pip安装速度,自己亲身体验~,原镜像下载速度不超过10K,阿里云镜像10M~~就是这样的差距

7)打开Anaconda Prompt,输入:

conda activate tensorflow

进入刚刚自己创建的环境里面

8)再输入:

pip install --ignore-installed --upgrade tensorflow 

9)测试:

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))

 

5.安装GPU版本的TensorFlow

前三步和CPU版本的一样

4)新建环境:tensorflow-gpu

5和6步不用再做了

7)打开Anaconda Prompt,输入:

conda activate tensorflow-gpu

8)再输入:

pip install --ignore-installed --upgrade tensorflow-gpu

9)遇到错误:

https://github.com/GoogleCloudPlatform/google-cloud-python/issues/3884

Command "python setup.py egg_info" failed with error......
(Command "python setup.py egg_info" failed with error code 1 in C:\Users\sanja\AppData\Local\Temp\pip-build-ppc44asr\googleapis-common-protos\ #3884)

运行命令:

pip install --upgrade setuptools

10)遇到错误:

https://github.com/tensorflow/tensorflow/issues/17101

ImportError: Could not find 'cudart64_90.dll'. #17101

这是因为目前tensorflow只支持cuda9.0,我开始安装的cuda9.2,所以出现这个错误

卸载cuda9.2安装cuda9.0解决

安装后还不行就重启,注意环境变量里面添加cuda9.0的路径,一般是自动添加的

 

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

深山里的小白羊

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值