tensorflow-gpu

1、Cuda

查看Cuda支持的GPU型号:https://developer.nvidia.com/cuda-gpus

下载cuda:https://developer.nvidia.com/cuda-downloads

下载好之后推荐安装就好了。

一般默认安装在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

将Cuda安装目录下的bin文件夹和lib\x64文件夹添加到path。

2、Cudnn

根据Cuda版本下载对应Cudnn:https://developer.nvidia.com/rdp/cudnn-archive

下载好后解压如图:

将上面三个文件夹里的东西分别放到cuda安装目录同名文件夹下面

3、tensorflow

pip uninstall tensorflow

pip install tensorflow-gpu==1.11.0

注意:使用pip是不能同时打开jupyter notebook,否则会失败。

 

posted @ 2019-02-23 11:14  Austin_anheqiao  阅读(160)  评论(0)    收藏  举报