windows系统,搭建anaconda下的pytorch环境,并在VScode下使用
2024-01-03 16:04:27
使用windows系统,搭建anaconda下的pytorch环境,并在VScode下使用
一、创建anaconda环境
- 这个只需要按照下面这个链接下载安装并配置环境即可
https://blog.csdn.net/fan18317517352/article/details/123035625
二、在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 好像不能使用
五、参考链接
- anaconda安装cudatoolkit参考链接:https://blog.csdn.net/qq_41719643/article/details/109777042
- pytorch官网:https://pytorch.org/get-started/previous-versions/
- 打开pyhton path的教程:https://zhuanlan.zhihu.com/p/367740437
- 打开shell windows的教程:https://blog.csdn.net/R_Young/article/details/123162047
文章来源:https://blog.csdn.net/shilichangtin/article/details/135363028
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。 如若内容造成侵权/违法违规/事实不符,请联系我的编程经验分享网邮箱:veading@qq.com进行投诉反馈,一经查实,立即删除!