-
打开Anaconda Prompt
-
在cmd中依次输入以下代码来添加清华镜像源
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安装/升级pip
conda install pip
- 查看当前环境
conda info --envs
注:本文安装在默认环境【base】中,如有需要可以新建其他环境并切换至该环境,代码如下:
conda create -n pytorch python=3.8
conda activate pytorch
conda info --envs
即可看到 conda 环境中,有新建的 pytorch 环境,右边的 * 号表示,当前你处于哪个环境。
- 在Pytorch网站查询需要安装的版本(点击跳转),如下:
- 复制安装代码,注意去掉后缀,如下。
conda install pytorch torchvision torchaudio cudatoolkit=10.2
-
弹出提示,输入 y,即可完成安装,显示“done”。
-
测试是否安装成功:
(1)输入:conda.bat activate pytorch (不报错)
(2)输入:python (显示版本号,如Python 3.7.7)
(3)输入:import torch(不报错)
(4)输入:torch.cuda.is_available(显示<function is_available … …>)
完成以上(1)至(4),说明安装成功。
如果本文对你有用,请点个赞加个关注吧~