安装过程很曲折······
首先,Python必须是3.5-3.7版本的:
具体见TensorFlow官网
而现在的Anaconda的Python是3.8版本的:
这可能是我之前下载TensorFlow库一直失败的原因之一。(看官方网站/文档很重要!)
所以,我在清华大学开源软件镜像站上下载Anaconda的4.2.0版本
4.2.0版本的Anaconda用的Python是3.5.2版本的。
然后,安装CUDA和cuDNN:
具体的安装过程和环境变量的配置略,有很多博客都说得很清楚了。
需要注意的是,CUDA一定要选择10.0版本的,然后cuDNN也一定要选对应于CUDA 10.0的。
最后,安装TensorFlow:
看清自己的电脑能否安装tensorflow-gpu。
打开cmd安装tensorflow-gpu:
pip install tensorflow-gpu==2.0.0 -i https://pypi.douban.com/simple
但我遇到了这个错误:
即:Cannot remove entries from nonexistent file d:\codingtools\anaconda3\lib\site-packages\easy-install.pth
我打开它所说得d:\codingtools\anaconda3\lib\site-packages路径,下面并没有easy-install.pth这个文件,于是创建了一个,然后执行如下指令:
pip uninstall setuptools
pip install tensorflow-gpu==2.0.0 -i https://pypi.douban.com/simple
然后就安装成功了:
打开ipython测试一下:
测试用的py程序:
import tensorflow as tf
import os
# os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
# os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
print('GPU:', tf.test.is_gpu_available())
a = tf.constant(2.)
b = tf.constant(3.)
print(a+b)
print(a*b)