由于换电脑了,原电脑显卡太垃圾,跑不了GPU版本就一直没装,其实安装过程蛮简单的,只要按步骤来即可。
目录
五、在JupyterNotebook中配置tensorflow
一、安装Anaconda
Anaconda安装地址:官方下载地址
这一步傻瓜式操作即可,但注意需要配置Anaconda环境
激活一个虚拟环境
conda create -n TF_2C python=3.6
# 进入虚拟环境
conda activate TF_2G
二、安装cuda和cudnn
cudnn其实就是cuda的一个补丁
注意安装cuda10.0的版本,tensorflow还不支持11.0以上的cuda,cudnn选择cuda匹配的即可。
cuda下载地址:官方下载地址
cudnn下载地址:官方下载地址
- cuda安装傻瓜式操作,注意选择自定义安装,其中不要选择以下几项。
CUDA-》Visual Studio Integeration / Nsight Compute
- cuda安装完毕后解压cudnn
- 将所有目录复制到cuda目录中,完整目录如下
- 配置环境变量
三、安装Tensorflow GPU版本
pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple # 注意报错就多装几次
四、测试是否安装成功
#以下是在python环境中
import tensorflow as tf
tf.test.is_gpu_available()
# 返回结果为True则安装成功
五、在JupyterNotebook中配置tensorflow
# 激活TF环境
conda activate TF_2G
# 安装jupyter和ipython
conda install ipython
conda install jupyternotebook
# 测试内核
jupyter notebook
# 在对应窗口输入
import tensorflow as tf