Windows环境下Tensorflow_gpu相关环境搭建
参考视频:https://www.bilibili.com/video/BV1Lv41177BW
1.下载CUDA
明确自己需要的tensorflow_gpu版本,找到对应的CUDA和cdDNN。
https://tensorflow.google.cn/install/source_windows#gpu,此链接中可以找到相关信息
如:tensorflow_gpu-1.14.0,需要的cuDNN为7.4,CUDA10。
在下载CUDA10之前还需要知道自己电脑的显卡驱动是否支持。
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html,此链接可以找到相关信息
若不支持,则需要升级显卡驱动。
CUDA下载地址如下:
https://developer.nvidia.com/cuda-toolkit-archive
下载与安装过程就不赘述,请参考视频。
CUDA安装好之后我发现终端输入 nvcc -V 并没有出现有效信息,须在系统变量中添加路径,参考此博文之后解决:https://blog.youkuaiyun.com/sinat_23619409/article/details/84202651
2.下载cuDNN
参考视频下载即可
51
2.下载cuDNN
参考视频下载即可