【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文件中,比如我的就在在这里插入图片描述中。
在这里插入图片描述

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

ZoomToday

给作者倒一杯卡布奇诺

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

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

打赏作者

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

抵扣说明:

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

余额充值