首先不要安装特别新的cuda,tensorflow版本,容易出现版本不匹配问题。
1.一定要确认有n卡且打开独显,并检查显卡驱动支持的最大cuda版本(网上有步骤)
2.切记用pip install tensorflow-gpu。不要用conda,因为哪怕指定gpu会自动装CPU版的,在运行时计算机可能默认使用cpu而压根检测不到你的gpu。如果pip太慢,使用清华等换源代码示例:
pip install -i https: //pypi. tuna. tsinghua. edu. cn/simple tensorflow-gpu==1. 14.0
3.cuda可以装在c盘以外,亲测可以
4.环境变量不区分大小写,无论新建pathPathPATH都会覆盖掉以前的环境变量,就算覆盖了,去360系统修复系统默认环境变量
5.numpy版本如果和tensorflow版本不对应也可能会报tensorflow和cuda的错…很迷惑
6.版本吧,这个没有办法
检测方法
1.测试tensorflow能不能导入,以及numpy等版本,导入就行,版本会报错中有version
2.cmd中nvcc-V检测cuda是否安装成功,显示版本号就是已成功
3.测试能不能检测到gpu
gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
cpus = tf.config.experimental.list_physical_devices(device_type='CPU')print(gpus)
print(gpus,cpus)
4.tf.test.is_gpu_available()检测能不能用
想到再改吧。
版权声明:本文为CSDN博主「朱彦霖12345」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_30090311/article/details/113834007
暂无评论