关于安装pytorch

前置：

默认已经安装好python和anaconda，并配置好环境

1、查看电脑当前配置

使用命令nvidia-smi，

nvidia-smi

---

确定可以安装的最高版本的cuda为12.0，所以要下载版本低于12.0的cuda。

2、进入anaconda虚拟环境

a、打开anaconda

b、若无虚拟环境，创建（有则跳过该步）
conda create -n “环境名”（文章创建的名字为pytorch）python=版本号（此处为3.9），记得带上版本号，不写版本号后续会报错，可能是因为电脑原本安装的python版本可能对不上建立虚拟环境的python版本，导致找不到，输入指令”where python”后没有输出。

conda create -n pytorch python=3.9

c、进入虚拟环境（此步很重要）
查看有已经哪些虚拟环境，conda info -e或conda env list，

conda info -e
conda env list

激活虚拟环境，activate 环境名（此处为pytorch）

conda activate pytorch

3、安装pytorch

去官网https://pytorch.org/

选择版本，11.8的cuda（用gpu的，不用的话则下载cpu版本），直接复制下面连链接。

将命令复制，粘贴到命令行窗口，回车（此处已经安装）。
等待安装，此处命令中有-c pytorch，是用的官方的源，删了则是用自己的源。官方的源不科学上网也可以在1h内下好。如果出现了报错问题，报错信息说retry的，可能是网络有问题，可以重新下载。
更新：2025.1.2，无论是否科学上网，用上述指令不能成功下载。使用pip指令下载可以成功，
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118

4、安装完成

会显示done
输入pip list，可以查看已经安装好的包。
输入pip list | grep torch，可以查看是否有torch的包。
输入python，进入python环境，输入import torch，不报错说明安装成功。

pip list
pip list | grep torch

输入torch.cuda.is_available()查看torch是否可以使用显卡，True就代表可以！

5、将虚拟环境导入python解释器

 
找到添加解释器，

将pytorch中的python.exe文件路径加入即可。
把使用先有环境改为创建的虚拟环境。

注：若显示conda找不到可执行文件，先输入anaconda 的Script中的conda.exe文件路径，点击右边的加载文件，即可将pytorch中的python.exe文件路径加入即可。最后选择使用pytorch环境，完成导入。

6、其他

Conda虚拟环境打包迁移

# 打包
conda activate project
conda env export > envirment.yml

# 创建
conda env create -f envirment.yml -n new_env