Win0搭建的环境是Python3.6.5 + Cuda10.0 + Cudnn7.6.3 + Tensorflow1.15.0,可参考。
注:
Cuda,Cudnn,Tensorflow的版本是相互对应。
我的Ubuntu装的是Cuda9.0 + Cudnn7 + Nvidia384 + Tensorflow1.8.0
由于我想使用的是Tensorflow1.14.0和Tensorflow1.15.0,百度tensorflow和cuda版本对应。
1.下载
Cuda:cuda_10.0.130_411.31_win10,
Cudnn:cudnn-10.0-windows10-x64-v7.6.3.30
>https://developer.nvidia.com/cuda-downloads
>https://developer.nvidia.com/rdp/cudnn-archive#a-collapse742-10
2.安装Cuda
双击
cuda_10.0.130_411.31_win10.exe
但是之后出现了安装失败,原因是因为Visual Studio(VS) Intergration无法安装
之后通过自定义的方式,取消Visual Studio Intergration进行安装,安装OK。
我暂时也不使用visual studio,所以我就没管,好像也没什么影响。
3.之后解压Cudnn的压缩包
cudnn-10.0-windows10-x64-v7.6.3.30.zip
若为默认安装,则应分别拷贝到的文件夹如下:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
4.nvidia-smi
之后就安装ok了,如果想使用nvidia-smi这个命令看的话,需要把它加入环渐变量中系统变量中path :C:\Program Files\NVIDIA Corporation\NVSMI
打开cmd,输入nvidia-smi
到这里就ok啦!