【嗐,终于搞好了】
【Anaconda3+tensorflow2.1】
1.下载Anaconda3-2019.10-Windows-x86_64 链接:https://pan.baidu.com/s/1kj-lcB6X4lc9YDLfbIXniQ 提取码:rbk
2. 打开Anaconda Promp
3.安装tensorflow2.1需要VS2019环境,安装 VC_redist.x64.exe
https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads
4. 依次输入以下命令,碰到y/n选项时选择y
1. conda create -n TF2.1 python=3.7
2. conda activate TF2.1
3. conda install cudatoolkit=10.1
4. conda install cudnn=7.6
5. pip install tensorflow==2.1
6. python
7. import tensorflow as tf
8. tf.__version__
若显示2.1.0则安装成功
5. 配置PyCharm
配置为TF2.1下的python
6. 运行测试代码
import tensorflow as tf
tensorflow_version = tf.__version__
gpu_available = tf.test.is_gpu_available()
print("tensorflow version:", tensorflow_version, "\tGPU available:", gpu_available)
a = tf.constant([1.0, 2.0], name="a")
b = tf.constant([1.0, 2.0], name="b")
result = tf.add(a, b, name="add")
print(result)
输出为以下为正确
tf.Tensor([2. 4.], shape=(2,), dtype=float32)
若报错 CUDA driver version is insufficient for CUDA runtime version
则下载自动更新软件更新显卡https://www.nvidia.cn/geforce/geforce-experience/