PyTorch安装排雷

CUDA安装

首先查看PyTorch官网适配的CUDA版本,选择适配的版本进行下载安装

cuDNN下载安装

登录NVIDIA需要Windows关闭防火墙,选择与CUDA同版本的cuDNN下载压缩包,解压后文件夹覆盖到CUDA文件夹

PyTorch安装

通过pip进行安装

pip --default-timeout=1000 install -U 模块名
下载安装包时,让它的延迟检测时间变长。

cmd输入:
pip --default-timeout=1000 install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
进行安装

原文地址:https://www.cnblogs.com/JinZL/p/13889047.html