tensorflow的升级与版本管理 – Maddock

1 查看cudacudnn的版本

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

图片[1]-tensorflow的升级与版本管理 – Maddock-唐朝资源网

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

图片[2]-tensorflow的升级与版本管理 – Maddock-唐朝资源网

6TF升级

1.我下载的是最新版本,以后如果需要新版本,比如升级到1.5.0

$—gpu==1.5.0

2.也可以登录看看有没有更新,然后先卸载,再更改对应位置,

但是老版本必须卸载,对应的cuda和cudnn必须对应

此环境已安装

7 次测试

你好=tf.(‘你好,’)

sess=tf.()

打印(sess.run(hello))

你好!

© 版权声明
THE END
喜欢就支持一下吧
点赞11赞赏 分享
评论 抢沙发
头像
欢迎您留下宝贵的见解!
提交
头像

昵称

取消
昵称表情代码图片

    暂无评论内容