windows环境下 pytorch离线安装

windows环境下 pytorch离线安装

 

1、问题描述：由于在下载依赖包时pytorch包非常大，无论是更改清华源、阿里源都下载的非常缓慢

（1）首先找到自己Windows显卡对应的 cuda版本

 找到：NVDIA控制面板-->帮助-->系统信息-->组件 ，如下图所示：对应版本为 cuda 11

 

（2）在pytorch官网  https://pytorch.org/get-started/locally/

 按照cuda版本 配置信息如下：

(3)  切换到对应conda环境下，执行命令： conda activate pytorch

（4）执行 Run this Command 中的命令：

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

该命令会先分析 需要用到的依赖，如下：

（5）在 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/ 

搜索并找到相关依赖 如： py3.8_cuda110_cudnn8_0 ，

 

（4）使用迅雷下载后，离线进行安装，命令如下：

conda install --offline   py3.8_cuda110_cudnn8_0