首先在cmd中使用
# 首先可以查询一下conda目前支持的cudatoolkit版本
conda search cudatoolkit --info
conda支持的版本如下,因为我们要装的pytorch是需要cuda > 11.0, 选择cuda=11.7或者11.8的,因此只截屏了我们将要安装的cudatoolkit
pytorch的版本选择只需要满足cuda=11.7或者cuda=11.8的最新的,不需要一定是下面图片pytorch
# 再创建一个虚拟环境
conda create -n lvxiaoleyolo python=3.10.12
# 这里写python=3.10.12, 如果报错的话就写python=3.10
在这里需要安装cuda,我们只需要给我们自己创建的虚拟环境安装cudatoolkit即可
# 查询是虚拟环境否创建成功
conda info --envs
# 激活虚拟环境
activate lvxiaoleyolo
# 直接安装跟pytorch对应的虚拟环境即可
conda install cudatoolkit==11.8.0
# 然后安装pytorch
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
# 查看一下安装的pytorch是不是支持cuda
python
import torch
torch.__version__
torch.cuda.is_available()
直接点击VScode右下角的
然后会弹出这个窗口,直接选择就可以
首先安装下面图片的,“点击左下角设置——>setting”
然后搜索python path
, 找到下列图片中的位置,并将其修改为想要使用的anaconda创建的虚拟环境
然后就会得到下图右下角的python运行环境,但是注意,此时使用终端运行程序的时候,仍然不能使用activate lvxiaoleyolo
激活虚拟环境
此时,需要回到步骤2,搜索shell windows
,然后根据下面图片选择Comand Prompt
pip install -r requirements.txt
# conda install -r requirements.txt 好像不能使用