> 文章列表 > pytorch安装(解决网络慢)

pytorch安装(解决网络慢)

pytorch安装(解决网络慢)

一、确定pytorch版本

# CUDA 11.6
pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116

二、解决网速过慢

1、下载(链接)到本地

  • 需要下载:torch、torchaudio、torchvision 
  • 下载的版本如下所示
    • cu116:cuda 11.6
    • cp37:python=3.7

2、安装

pip install ./"torch-1.12.0+cu116-cp37-cp37m-win_amd64.whl"pip install ./"torchaudio-0.12.0+cu116-cp37-cp37m-win_amd64.whl"pip install ./"torchvision-0.13.0+cu116-cp37-cp37m-win_amd64.whl"

参考:[PyTorch] 安装笔记, 基于Windows10/cuda11.6 - 知乎 (zhihu.com) 

到此,pytorch安装完毕!!!