tensorflow-gpu安装参考:
https://www.cnblogs.com/yincq/p/11603057.html
然后跑程序的时候发现还是用的cpu
测试了以下两行代码:
import tensorflow as tf
tf.test_is_gpu_available()
发现缺少了几个文件
在这里插入图片描述
解决方法:
在这里插入图片描述
复制bin目录下的curand64_100.dll文件,粘贴,重命名为curand64_10.dll,其他的文件也这样操作,就可以使用gpu了。
在这里插入图片描述

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐