linux系统下,在anaconda虚拟环境中安装tensorflow-gpu
·
参考:
查看python,cuDNN,CUDA,Trnsorflow的对应关系,网址如下:https://tensorflow.google.cn/install/source?hl=zh-cn#gpu
eg:
# 1.创建虚拟环境
conda create -n tensorflow python=3.8
# 2.查看下载源中的包的各个版本,根据包含的版本号,指定下载版本
## -c conda-forge,指定搜索渠道conda-forge,查看包含的软件包的版本
conda search tensorflow-gpu -c conda-forge
conda search cudnn -c conda-forge
conda search cudatoolkit -c conda-forge
# 3.安装指定版本的tensorflow-gpu
## conda会自动安装匹配tensorflow版本的cudatoolkit和cudnn,不需要手动安装
conda install tensorflow-gpu==2.6.0 -c conda-forge
# 4.检测是否安装成功
import tensorflow as tf # 导入成功,说明tensorflow下载正确
print(tf.test.is_gpu_available()) # 返回True,则GPU可用
更多推荐


所有评论(0)