自己都不一定看得懂的随手记

 

 

更新显卡驱动

lspci | grep -i nvidia

查看显卡信息

https://www.nvidia.cn/Download/index.aspx?lang=cn

下载对应驱动，上传

卸载原来的驱动

sudo /usr/bin/nvidia-uninstall

sudo sh NVIDIA-Linux-x86_64-410.129-diagnostic.run

 

下载cuda

https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal

cudnn

https://developer.nvidia.com/rdp/cudnn-download

 

pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html