1、安装软件
cuda_10.0.130.1_linux.run
cudnn-10.0-linux-x64-v7.6.4.38.tgz
torch-1.2.0-cp36-cp36m-win_amd64.whl
torchvision-0.4.0-cp36-cp36m-win_amd64.whl
切忌一点:
linux下显卡要单独安装,不要用cuda自带的显卡驱动。
windows下可以不用单独安装显卡驱动,用cuda自带的显卡驱动就可以。
参考:
ubuntu16.04安装cuda9.0+cudnn7.3.1+anaconda3+tensorflow1.12
windows10安装GPU版pytorch步骤(全程离线)
真实机下 ubuntu 18.04 安装GPU +CUDA+cuDNN 以及其版本选择