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

发布时间:2024年01月03日

一、创建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 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运行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 好像不能使用
    

五、参考链接

文章来源:https://blog.csdn.net/shilichangtin/article/details/135363028
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。