Pytorch官网
选择你要用的版本
由于国内直接安装pytorch会很慢,而且容易出错,所以一般会采用清华镜像安装速度会快很多。详细步骤直接引用自这里:https://mirror.tuna.tsinghua.edu.cn/help/anaconda/
在“ C:\Users\用户名 ” 目录下找到.condarc文件,将下面的内容拷贝到这个文件。如果没有这个文件,可先执行 conda config --set show_channel_urls yes 生成该文件之后再修改。
channels:
- defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
从pytorch官网找到对应版本的安装命令。这里我安装的是pytorch1.7,windows 10,cuda 10.1,python版本,在Anaconda power shell中执行下面的命令:
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
等几分钟就安装好啦!
要是你原来安装过可能会报错,可能是因为之前安装失败残留的文件。可以按提示删除C:\Users\Administrator\anaconda3\pkgs下的对应文件或文件夹,然后再次执行安装命令即可。
InvalidArchiveError(‘Error with archive
C:\Users\Administrator\anaconda3\pkgs\torchvision-0.8.1-py38_cu101.tar.bz2.
You probably need to delete and re-download or re-create this file.
Message from libarchive was:\n\nCould not unlink’)InvalidArchiveError(‘Error with archive
C:\Users\Administrator\anaconda3\pkgs\pytorch-1.7.0-py3.8_cuda101_cudnn7_0.tar.bz2.
You probably need to delete and re-download or re-create this file.
Message from libarchive was:\n\nCould not unlink’)