安装 :win10 + CUDA 11.0 + cuDNN8.0.2 + Pytorch-gpu
前言
注意: 点进来的各位,你觉得这是一篇正常的教程篇吗?git
不是的! 这是一篇劝退文。github
笔者安装cuda11+pytorch-gpu失败,故有此文。windows
安装时间:2020-08-07ui
缘由
- 安装CUDA和cuDNN时,请先打开Pytorch-gpu安装命令连接查看Pytorch是否有发布支持cuda11.0的binaries版本。
结果以下图:
能够看到,此时并没有支持cuda11的binaries版本。
Pytorch论坛中给出了答案:
spa
https://discuss.pytorch.org/t/pytorch-with-cuda-11-compatibility/89254blog
Pytorch源里贴出了windows 支持cuda11的PR:
教程
https://github.com/pytorch/builder/pull/463图片
文末老哥劝一句,目前仍是安装cuda10.2吧,从源build真的烦。
get
附cuda和cudnn连接:it
cuDNN: https://developer.nvidia.com/rdp/cudnn-download CUDA10.2 : https://developer.nvidia.com/cuda-10.2-download-archive