windows系统,搭建anaconda下的pytorch环境,并在VScode下使用

一、创建anaconda环境

二、在anaconda中创建一个虚拟环境,并加入cuda,为了方便安装pytorch

  • 首先在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 yolo python=3.10.12
    # 这里写python=3.10.12, 如果报错的话就写python=3.10
    
  • 在这里需要安装cuda,我们只需要给我们自己创建的虚拟环境安装cudatoolkit即可

    # 查询是虚拟环境否创建成功
    conda info --envs
    
    # 激活虚拟环境
    activate yolo
    
    # 直接安装跟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运行anaconda程序

第一种方法

直接点击VScode右下角的
在这里插入图片描述

然后会弹出这个窗口,直接选择就可以
在这里插入图片描述

第二种方法

  • 首先安装下面图片的,“点击左下角设置——>setting”
    在这里插入图片描述

  • 然后搜索python path, 找到下列图片中的位置,并将其修改为想要使用的anaconda创建的虚拟环境
    在这里插入图片描述

如果debug代码时,不能使用powershell,就使用下面方法改为cmd

  • 然后就会得到下图右下角的python运行环境,但是注意,此时使用终端运行程序的时候,仍然不能使用activate lvxiaoleyolo激活虚拟环境
    在这里插入图片描述

  • 此时,需要回到步骤2,搜索shell windows,然后根据下面图片选择Comand Prompt
    在这里插入图片描述

四、创建yolov5环境

  • 直接使用VScode打开yolov5代码文件包
    pip install -r requirements.txt
    
    # conda install -r requirements.txt 好像不能使用
    

五、参考链接

  • 23
    点赞
  • 30
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值