由于国内直接安装pytorch会很慢,而且容易出错,所以一般会采用清华镜像安装速度会快很多。详细步骤直接引用自这里:https://mirror.tuna.tsinghua.edu.cn/help/anaconda/
步骤一:修改.condarc文件
在“ 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')