Fork me on GitHub

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

posted @ 2020-10-15 22:30  Lurchy  阅读(3161)  评论(0编辑  收藏  举报