记录下,整个过程卡在cudnn上,开始安装了cuda10.1,pytorch目前不支持(以后应该会),果断安装cuda10.0
1、操作系统:ubantu 18.04
2、cuda版本 :cuda_10.0.130_410.48_linux.run
3、cudnn版本:cudnn-10.0-linux-x64-v7.4.2.24.tgz(注意,就是卡在这里,cuda10版本只能对应7.4版)
4、torch版本:torch-1.1.0-cp36-cp36m-linux_x86_64.whl
记得 pip3 install torchvision
5、keras版本:2.2.4
6、tensorflow-gpu版本:1.14.0
整个过程注意就是版本一定要下载对应的