#1 安装anaconda
官网下载安装即可。python3.7版本
#2 安装CUDA CUDNN tensorflowgpu1.14 对应CUDNN 7.6.1 CUDA 10.0
注意tensorflow不同版本所需的CUDA及CUDA对应的CUDNN不同,注意版本匹配!
参考 https://blog.csdn.net/omodao1/article/details/83241074
https://developer.nvidia.com/cuda-gpus
https://developer.nvidia.com/cuda-toolkit-archive
https://developer.nvidia.com/rdp/cudnn-download
这里安装了CUDA10.0版本
快速安装即可
之后安装对应的CUDNN
官网下载安装包后解压,将得到的文件夹复制到CUDA10.0的根目录
#3. 安装tensorflow-gpu
打开Anaconda Prompt,进入Anaconda命令行管理界面。配置清华仓库镜,输入指令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
创建运行环境,输入指令:
conda create -n tensorflow-gpu python=3.7
新建一个名字叫“tensorflow-gpu”,python版本为3.的运行环境,此环境与Anaconda中其它环境隔离。输入“y“和回车后开始安装。
激活并进入环境,使后续指令在激活的环境中生效,输入指令:
conda activate tensorflow-gpu
升级pip到最新版,防止稍后的安装时,出现错误。输入指令:
python -m pip install --upgrade pip
安装tensorflow1.7.0及相应依赖包,输入指令:
pip install --ignore-installed --upgrade tensorflow-gpu
至此,tensorflow的安装完成。
#4 验证
在命令行中,先进入tensoflow-gpu,再进入python,并输入以下代码:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
输出:
b’ Hello, TensorFlow!
参考https://blog.csdn.net/weixin_39290638/article/details/80045236