安装 :win10 + CUDA 11.0 + cuDNN8.0.2 + Pytorch-gpu

冷暖自知 提交于 2020-08-09 16:24:56

安装 :win10 + CUDA 11.0 + cuDNN8.0.2 + Pytorch-gpu

前言

注意: 点进来的各位,你以为这是一篇正常的教程篇吗?

不是的! 这是一篇劝退文。

笔者安装cuda11+pytorch-gpu失败,故有此文。

安装时间:2020-08-07

原因

  1. 安装CUDA和cuDNN时,请先打开Pytorch-gpu安装命令链接查看Pytorch是否有发布支持cuda11.0的binaries版本。

结果如下图:
在这里插入图片描述
可以看到,此时并无支持cuda11的binaries版本。
Pytorch论坛中给出了答案:


https://discuss.pytorch.org/t/pytorch-with-cuda-11-compatibility/89254

在这里插入图片描述
Pytorch源里贴出了windows 支持cuda11的PR:

https://github.com/pytorch/builder/pull/463

在这里插入图片描述
文末老哥劝一句,目前还是安装cuda10.2吧,从源build真的烦。

附cuda和cudnn链接:

cuDNN: https://developer.nvidia.com/rdp/cudnn-download
CUDA10.2 : https://developer.nvidia.com/cuda-10.2-download-archive

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!