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

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

前言

注意: 點進來的各位,你覺得這是一篇正常的教程篇嗎?git

不是的! 這是一篇勸退文。github

筆者安裝cuda11+pytorch-gpu失敗,故有此文。windows

安裝時間:2020-08-07ui

緣由

  1. 安裝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

相關文章
相關標籤/搜索