pytorch安装教程gpu太慢

在安装PyTorch时，如果下载速度太慢，可以尝试以下方法[^1]：

1.使用清华镜像源安装。在命令行中输入以下命令，将清华镜像源加入conda的镜像源列表中。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

然后安装PyTorch。

conda install pytorch torchvision torchaudio -c pytorch

2.在官网下载whl文件安装。在官网上找到合适的whl文件下载链接，然后使用pip安装即可，例如：

pip3 install torch-1.7.0-cp36-cp36m-win_amd64.whl

如果要安装GPU版本的PyTorch，需要注意一下几点：
- 先确保已经安装好了CUDA和CUDNN。
- 需要安装与CUDA和CUDNN版本相对应的PyTorch版本。可以在官网上查看各个版本之间的对应关系。
- 安装时需要使用cuda和cudnn的版本号指定安装的版本，例如：

conda install pytorch==1.7.0 torchvision==0.8.0 cudatoolkit=10.1 -c pytorch