pytorch详细安装

打开annaconda自带的annaconda prompt

创建虚拟环境，这里以python=3.9.7为例。不知道自己安装的是什么版本的小伙伴可以去cmd
输入python查看，创建一个名为pytorch的虚拟环境

conda create -n pytorch python=3.9.7

进入虚拟环境安装pytorch

进入pytorch虚拟环境

conda activate pytorch

link:进入pytorch官网查看你要安装的版本号复制代码在虚拟环境中安装即可,这里以cuda 10.2为例，每个人的要安装的版本不一样，需要自己去查找

pip3 install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio===0.10.1+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

我们查看一下是否安装成功,在cmd里面依次输入

python

import torch

torch.cuda.is_available()

如果输出true则表示成功
false处理如下，删除安装的虚拟环境，重新运行一遍上面的操作

conda activate base
conda remove -n pytorch --all

把pytorch在annaconda prompt安装到annaconda 里面的jupyter notebook中

以管理员的形式打开annaconda prompt

conda install nb_conda

or

#首先安装ipykernel

conda install ipykernel

#在虚拟环境下创建kernel文件

conda install -n 环境名称 ipykernel

接着再进入前面创建好的虚拟环境里面输入

python -m ipykernel install --user --name pytorch

进入notebook查看

在新建文件那里我们看到有pytorch，并且打开后导入torch库，我们发现成功了！

import torch

删除虚拟环境在jupyter 上的 kernels

后续如果该环境不再需要的话可以按照以下步骤进行删除：

1 打开Anaconda Prompt
2 输入jupyter kernelspec list查看安装的kernel和位置
3 根据显示的路径进入其中把文件夹删除，重启jupyter notebook即可