微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

【TensorFlow】Win10下在Anaconda中安装创建GPU版的TensorFlow亲测,避坑

且看此处

说在前面

  使用的环境是win10+python3.7.10+tensorflow1.14.0gpu,仅供参考。

安装CUDA

  这里因为在之前安装GPU版本的PyTorch时已经安装好,所以我没有再进行安装CUDA。如果你没有安装CUDA的话,可以查看这篇文章进行下载安装,一定要注意自己电脑的Nvidia版本,选择合适的版本进行安装。

创建Anaconda虚拟环境

  参考这篇文章。如使用下面的命令进行创建,tf-gpu这个可以自定义,换成自己需要的名字。之后使用命令进入虚拟环境即可。

在这里插入图片描述

在这里插入图片描述

安装tensorflow-gpu

  进入虚拟环境后,输入命令pip install tensorflow-gpu==1.14,(这里可以指定版本,也可以不指定,认下载安装最新版的)。如果出现**.ReadTimeoutError: HTTPSConnectionPool(host=‘pypi.tuna.tsinghua.edu.cn’, port=443): Read timed out.**错误,更换下载源即可,比如pip install tensorflow_gpu==1.14 -i https://pypi.douban.com/simple/,在后面安装库时出现这个错误,都可以通过更换下载源解决。如果下载失败,多次尝试下载,更换下载源。

验证

  下载完成后,按下图进行验证。最后返回True说明安装成功了。

在这里插入图片描述

在这里插入图片描述

  或者使用下方的代码进行测试:

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c))

解决cublas64_11.dll not found,cublasLt64_11.dll not found,cufft64_10.dll not found,curand64_10.dll not found,cusolver64_10.dll not found,cudnn64_8.dll not found的问题,在验证时,如果出现上面的问题,解决方法有两种:
-直接在dll网站查找下载
-如果缺失的dll文件很多,或者有些dll文件在上面的网站中找不到,那么可能是tensorflow版本过高,无法适配,所以需要降低版本,卸载掉目前的版本pip uninstall tensorflow-gpu,再选择合适的版本进行安装。

Jupyter Notebook中使用内核

PyCharm中使用虚拟环境

  选择New Project,按下图找到你所创建环境中的python.exe文件即可。一般位于你创建环境目录下的envs文件中,比如我的就在

在这里插入图片描述中。

在这里插入图片描述

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐