- https://developer.nvidia.com/rdp/cudnn-archive (cudnn官网下载目录与cuda版本对应关系)
- https://developer.nvidia.com/cuda-toolkit-archive(cuda官网下载地址)
- https://tensorflow.google.cn/install/source_windows#gpu(tensorflow-gpu与cudnn,cuda版本对应查询)
https://mckayward.github.io/floyd-docs/guides/environments/(keras与tensorflow版本对应关系查询)
tensorflow-gpu
nvcc -V (查看cuda是否安装成功)
操作
-把cuDNN文件夹下面的bin,include,lib文件夹里面的内容复制到CUDA文件夹下面对应的文件夹内,一定要注意,然后再把CUDA文件夹下面extras\CUPTI\libx64下的cupti64_90.dll 拷贝到 bin目录
-在path中添加cuda的bin目录和lib目录下的x64目录。
安装
1.查看环境:conda info --envs
2.创建环境:conda create --name xxx python=3.6
3.切换环境:activate xxx
4.切回原来的Python环境:deactivate xxx
5.删除不要的环境: conda remove --name xxx --all
安装tensorflow-gpu1.4
pip install tensorflow-gpu==1.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
安装keras2.0.8
pip install keras==2.0.8 -i https://pypi.tuna.tsinghua.edu.cn/simple