已安装好Anaconda环境为前提
1.打开Anaconda Prompt
输入下面代码,查看conda环境下的内容:
conda info --envs
输入下面代码,新建pytorch环境,并给环境安装python
conda create -n pytorch python=3.8
切换到新建的pytorch环境中:
conda activate pytorch
2. 安装pytorch准备
(1)进入pytorch官网,https://pytorch.org/ ,根据自己的电脑选择相应配置
(2)更换清华镜像源,依次输入下列代码
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/
(3)复制pytorch页面的代码,如截图选中部分,我的是:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
去掉后面的 -c pytorch并运行。
(4)安装完成后验证是否安装成功
输入 pip list ,发现torch的版本
输入python,进入python环境。输入import torch 没有报错。
输入 torch.cuda.is_available() ,有返回true / false .