注意,需要科学上网:
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的话应该就没问题了。
不能科学上网的话,可能装不了。