复制下面链接到迅雷下载或者直接复制地址在网页里面打开:
CUDA10.0:https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_411.31_win10
下载完成CUDA10.0后进行安装:
选中自定义安装,然后一直下一步,直到结束
配置环境变量:
检查系统变量Path里有没有这一条C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin,没有的话添加进去
再检查下面的框是否有这两条:
CUDA_PATH C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
CUDA_PATH_V10_0 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
没有的话添加进去
nvcc -V验证一下是否已经装好。
出现上图说明CUDA已经安装好
cudnn 7.4:
https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.4.1.5/prod/10.0_20181108/cudnn-10.0-windows10-x64-v7.4.1.5.zip
官网可能需要注册账号下载,不过注册很简单,按要求来就行
下载好压缩包解压缩,把里面的3个文件夹:
复制到CUDA的文件夹下:
直接覆盖就行
然后在系统环境变量的Path中,加以下路径
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64
这样就把cudnn也弄好了
打开anaconda prompt
conda create -n tf1 python=3.6.9 (tf1是自己起的环境名,可以改成自己喜欢的)
按y安装完会出现下图:(应该是tf1,这是创建tf2环境时截的图)
根据提示,
输入conda activate tf1 激活环境
(提示的第二个conda deactivate 是退出激活的环境)
出现红框标的环境名,证明环境创建激活成功。
接着进行下面两步:
conda install jupyter notebook # 安装ipykernel模块
python -m ipykernel install --user --name tf1 --display-name "tf1" # 进行配置
完成以后会发现已经可以在jupyter notebook里使用:
pip install tensorflow-gpu==1.14.0
安装完成后,检查是否成功:
或者打开jupyter notebook检查:
import tensorflow as tf
print('Tensorflow Version:{}'.format(tf.__version__))
print(tf.test.is_gpu_available())
到这里tensorflow 1.14就安装完成了
步骤和安装1.0几乎一样
1.安装CUDA11.0+cudnn8.0
流程和一相同
配置环境时改一下版本号就可以
2.用anaconda创建虚拟环境tf2
conda create -n tf2 python=3.6.9 # 创建一个python3的环境,名为tf2
conda activate tf2 # 激活tf2环境
conda install jupyter notebook #安装ipykernel模块
python -m ipykernel install --user --name tf2 --display-name "tf2" #进行配置
完成以后会发现已经可以在jupyter notebook里使用:
pip install tensorflow-gpu==2.4.0
检查是否安装成功:
更改系统环境变量,CUDA_PATH的v10.0/v11.0即可
用tf1就改成v10.0;用tf2就改成v11.0。
报错:AttributeError: module 'numpy' has no attribute 'integer',解决如下:
卸载当前numpy
pip uninstall numpy
下载对应版本1.16.0
pip install -U numpy=1.16.0 (对应tensorflow1.14版本)