安装python版本(可省略)
应该安装 python 3.5-3.7
1 创建虚拟环境并激活
conda create -n Your_name python=3.6
conda activate Your_name
2 安装VC++
https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads
名字 vc_redist.x64.exe
安装完成后重启!!
3 conda安装cuda
版本2
conda install cudatoolkit=10.1
必须是10.1
版本1 tensorflow 1.8对应cuda 9
conda install cudatoolkit=9.0
4 conda 安装cudnn
版本2
conda install cudnn=7.6.5
版本1
conda install cudnn=7
版本号不小于7.6
5 TensorFlow安装
版本2
pip install tensorflow-gpu==2.3.0 -i https://pypi.douban.com/simple/
版本1
pip install tensorflow-gpu==1.8 -i https://pypi.douban.com/simple/
6 测试
import tensorflow as tf
print(tf.test.is_gpu_available())
解决TensorFlow1安装中出现numpy问题
pip install --user numpy==1.16.4 -i https://pypi.douban.com/simple/
看到True就是安装成功了。
取消红字警告
import os
#屏蔽红字警告
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'