栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

ubuntu,conda安装tensorflow-gpu老版本

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

ubuntu,conda安装tensorflow-gpu老版本

注意,需要科学上网: 1.如果之前有添加过conda镜像源,请删除镜像源
conda config --remove-key channels
2.显示tensorflow-gpu可安装的库
anaconda search -t conda tensorflow-gpu

或者进入https://anaconda.org/
搜索tensorflow-gpu,点进去后输入命令即可,不过不推荐,因为不会显示它拥有的版本
例如anaconda官方的

 conda install -c anaconda tensorflow-gpu 
3.查看每个库拥有的版本,这边选择anaconda官方的
anaconda show anaconda/tensorflow-gpu

会显示1.0-1.15以及2.0-2.5版本的tensorflow

4.安装tensorflow-gpu,这边选择1.14,也可以选择其他版本
conda install --channel https://conda.anaconda.org/anaconda tensorflow-gpu=1.14
5.注意事项

如果直接安装anaconda官方版本的话,输入4.的命令即可,一定要把之前添加的镜像源删了,要不然安装不了指定版本,如果添加了清华镜像的话,会装上1.3的,测试cuda也能用。
我这边环境是3090,然后本地装了个cuda11.1,用pip装的话只能装上 2.5,2.6版本的,cuda能用。用conda装的话其实不用在本地装cuda,因为和pytorch一样,他会装个cudatoolkit,其实就相当于cuda了,不过不像pytorch那样要强匹配,比如cudatoolkit=11.1,pytorch就只能装1.8和1.9。装anaconda官方的1.15版本gpu用不了,试了1.14和1.7能用,其他的没试。
测试代码tf.test.is_gpu_available(),返回true的话应该就没问题了。
不能科学上网的话,可能装不了。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/339374.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号