天天看点

Anaconda3+tensorflow2.1+Pycharm安装与配置【Anaconda3+tensorflow2.1】

【嗐,终于搞好了】

【Anaconda3+tensorflow2.1】

1.下载Anaconda3-2019.10-Windows-x86_64        链接:https://pan.baidu.com/s/1kj-lcB6X4lc9YDLfbIXniQ 提取码:rbk

Anaconda3+tensorflow2.1+Pycharm安装与配置【Anaconda3+tensorflow2.1】

2. 打开Anaconda Promp

Anaconda3+tensorflow2.1+Pycharm安装与配置【Anaconda3+tensorflow2.1】

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

Anaconda3+tensorflow2.1+Pycharm安装与配置【Anaconda3+tensorflow2.1】

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/