pytorch安装（anaconda）

2022-02-28发布

本人之前已经安装过anaconda了，读者请自行下载安装anaconda，链接如下

Anaconda | Individual Edition

下面开始pytorch的安装

直接安装也行，为了更多的学习，我选择了新建一个虚拟环境，不想麻烦可以跳过前面这几步（虽然我没试过直接装）

准备虚拟环境

网上查阅了听说python3.6版本的兼容性很好，所以开始安装3.6

如上点击进入，运行以下命令，py36为虚拟环境名称，=3.6为python版本为3.6，遇到y/n选项直接输入y回车

conda create -n py36 python=3.6

安装完输入

activate py36

进入环境，有base转为py36如图

虚拟环境安装完，安装pytorch

进入pytorch官网：Start Locally | PyTorch

 选择GPU（如果有独立显卡），无则选择CPU

查看该怎么选择

1、任务管理器->性能可以看到GPU（如图）

 2、打开cmd输入以下命令得到箭头所示大于390即可选择GPU

nvidia-smi

选择好在新建的虚拟环境中运行（根据自己电脑选择）遇到y/n输入n等待安装完成（中间若安装失败重新运行相同命令即可）

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

 补充，若cmd输入nvidia-smi没反应则在系统环境变量Path中添加即可

C:\Program Files\NVIDIA Corporation\NVSMI

确认是否安装成功

在虚拟环境中依次输入python->import torch->torch.cuda.is_available()

输出true即安装成功且可以使用GPU（如图所示）

新的虚拟环境在jupyter中使用的方法

在进入py36虚拟环境如图

 继续输入

conda install ipykernel

安装完然后conda环境注入

python -m ipykernel install --user --name your_env --display-name "your_display"

针对我的即python -m ipykernel install --user --name py36 --display-name "Python (py36)"

your_env 是你的conda环境名称，我的就是py36
your_display：将是你在notebook中看到的名称。我写的是Python(py36)

然后打开jupyter得到如图

到此结束

补充一些命令（Anaconda Prompt (anaconda3)中运行）

查看所有环境 conda env list

进入环境 activate py36        //py36为我自己的环境，根据实际修改     

回到base环境   activate root

删除环境  conda remove

查看已有的kernel （查看jupyter环境）        jupyter kernelspec list

删除已有的kernel     jupyter kernelspec remove kernelname

2023.10.20补充：

cuda下载

 CUDA10.1下载地址

参考：win10下pytorch和CUDA的安装完整过程_install pytorch torchvision cudatoolkit=10.1 -c py-CSDN博客

 下载速度慢，推荐使用离线下载，可参考：

pytorch下载太慢的解决办法-CSDN博客