Ubuntu16.04+3090+cuda11.0+cudnnV8+pytorch-nightly

参考博客: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也写不入数据:)