步骤1
下载anaconda,配置anaconda环境。
步骤2
前往pytorch官网,根据电脑配置生成指令。
步骤3:
软件安装过大,可以使用清华源进行安装。在anaconda promote中执行以下指令:
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
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
之后在命令行中输入官网生成的指令
conda install pytorch torchvision torchaudio cudatoolkit=11.3
注意要去掉 -c pytorch
可以使用pip list查看是否安装成功
查看cuda是否安装成功