cuda一般安装在/usr/local/cuda/路径下,这个路径下有一个.txt文件,里面记录了cuda的版本信息
cat /usr/local/cuda/.txt 来查询
同理,cudnn的信息在它的头文件中
cat /usr/local/cuda//cudnn.h | grep -A 2 查询
2对应cuda和cudnn的版本
CUDA 8.0→cuDNN v5.1
CUDA 8.0→cuDNN v6.0
CUDA 9.0→cuDNN v7.0.5
1.6/1.5对应CUDA 9.0,
1.4/1.3对应CUDA 8.0
3 查看版本
不带gpu指cpu版本
sudo pip show -gpu
sudo pip 显示
4 卸载
sudo pip -gpu
sudo pip
5 安装指定版本
sudo pip -gpu==1.4.0
或
sudo pip -U –pre -gpu==1.4.0
6TF升级
1.我下载的是最新版本,以后如果需要新版本,比如升级到1.5.0
$—gpu==1.5.0
2.也可以登录看看有没有更新,然后先卸载,再更改对应位置,
但是老版本必须卸载,对应的cuda和cudnn必须对应
此环境已安装
7 次测试
你好=tf.(‘你好,’)
sess=tf.()
打印(sess.run(hello))
你好!
© 版权声明
本站下载的源码均来自公开网络收集转发二次开发而来,
若侵犯了您的合法权益,请来信通知我们1413333033@qq.com,
我们会及时删除,给您带来的不便,我们深表歉意。
下载用户仅供学习交流,若使用商业用途,请购买正版授权,否则产生的一切后果将由下载用户自行承担,访问及下载者下载默认同意本站声明的免责申明,请合理使用切勿商用。
THE END
暂无评论内容