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
进行安装