參考博客:https://blog.csdn.net/qq_33200967/article/details/80689543
(夜雨飄零nb)
驅動的另一種安裝方式:https://blog.csdn.net/qq_43030766/article/details/91382868
安裝驅動版本為11.1或者11.0
卸載cuda,
安裝cuda11.0
只需把cuda-10.0改成cuda-11.0
安裝cudnn v8
cudnn安裝參考博客:https://blog.csdn.net/Thanlon/article/details/108279649#2_cuDNN_25
anaconda安裝pytorch
conda install pytorch torchvision cudatoolkit=11.0 -c pytorch-nightly
無法換源
靜等下載完成即可
如果你的顯卡是3090,只能安裝pytorch-nightly
安裝其他即使成功,gpu也寫不入數據:)