MissWangMou的博客

原创 cuda版pytorch兼容问题

在下载GPU版pytorch前必须先下载cuda和cudnn。建议cuda11.3+cudnn8.2.1.3+pytorch1.10(目前官网首页)，这样我运行下来没有任何问题。 下载cuda 先看电脑显卡支持的cuda版本，注意：这里查到的是最高版本，你可以选择下载低版本。 win+R 输入cmd 打开窗口后输入：nvidia-smi 我的最高版本是11.6 ，那么11.6前的都可以下载 我下载的11.3 链接：​​​​​​CUDA Toolkit 11.3 Downloads | NVID..