一、Anaconda安装
跳过
二、安装环境
1. 安装什么环境——根据CUDA版本去官网查
首先,用Pytorch无非是想使用自己电脑的GPU进行计算加速,那我们就要知道这电脑上的显卡的Cuda版本,以下面为例,使用以下指令可以查询到cuda版本:
nvidia-smi
可以看到CUDA版本是12.1,根据这个CUDA,我们要到Pytorch的官网上查询我们要下载torch和torchvision(又或者是torchaudio)。
网址是:download.pytorch.org/whl/torch_stable.html
在这个例子里,只要你安装的torch的cuda版本是低于12.1的就没有问题,比如之前我安装的一个环境就能成功运行:
不过最稳妥的方法还是应该安装对应CUDA版本的torch,不过我们这里就不要求了,就选取11.8版本的就ok:
可以看到在11.8版本下有很多版本的torch和对应适配的python版本,如果我们选择安装torch2.0.1版本那么可以选择3.8-3.11版本的python。
于是查看完毕之后,明确目标:cu118+python3.8+torch2.0.1+torchvision0.15.2
2. 怎么安装环境——conda+pip
首先,创建一个名为test的环境(为了防止重名,可以conda env list去查看已有的环境列表):
conda create -n test python=3.8
这里在向你确认安装环境所必须的包,输入y然后回车就行。如果下载太慢可以换成清华源:(最近清华源有点问题!建议还是先别用,老老实实官网下)
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
(如果你不想把所有的下载都设置在清华源上也可以尝试后面说的第二种方法。)
然后我们激活环境,并用刚刚安装好的pip去安装我们需要的torch和torchvision:
conda activate test
pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 -f https://download.pytorch.org/whl/torch_stable.html
使用其他源的另一种方法是紧接着在上述pip语句之后加:
-i https://pypi.tuna.tsinghua.edu.cn/simple
于是就可以快乐下载安装了。至于安装python的包,requirement.txt那些就不说了。
安装成功之后,在python输入以下语句就可以查看是不是装成功了:
三、复制环境,删除环境,环境名更改
这篇文章写的很好:Conda常用命令
四、将创建的解释器(虚拟环境)在VScode里面使用
安装VScode之后,一般要装这些扩展包:Chinese-Simplified、Python、Remote-ssh。
然而在装了python之后,很有可能还是提示你选择解释器。
此时就涉及到找解释器的问题,其实我们再Anaconda Prompt里面输入conda env list就可以看到了:
此时运行程序,极有可能出现以下的桥段:
此时你就要想一下有没有把Conda的一些东西加到环境变量里了。