装了大半天的GPU,记录下。
亲测CUDA10.1是不行的,python3.7用不了。
配置:
1.win10
2. Anaconda3-2019.03-windows-x86_64
3. python3.7.3
4. tensorflow-GPU版本:1.14.0
5. CUDA 10.0
6. cuDNN v7.6.1
安装步骤:
一、安装python和anaconda,记得设置环境变量
二、安装cuda,设置环境变量
三、解压cuDNN,把里面的文件夹复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0(CUDA的安装路径)
四、安装tensorflow-gpu
在anaconda开启 anaconda prompt,运行下列命令:
conda create --name tensorflow114
activate tensorflow114
pip install tensorflow-gpu==1.14.0
五、检查安装情况
import tensorflow as tf
tf.__version__
安装完后马上跑程序可能依旧不行,这个时候把pycharm之类软件重启一下就可以了。