【已完成】windows配置pytorch2.4.1深度学习环境
前言
目前正在学习pytorch环境下学习yolo11中的语义分割部分，第一步就是要开始配置所需要的环境，在这边分享一下过程。
首先说明一下我的电脑配置：显卡是A5000，Windows10，需要装的是pytorch2.4.1 、cuda11.8和配套的cudnn、Python3.10.14其他按照yolo11中的需求安装。
第一步
根据显卡的需求选择要下载的cuda版本，注意：你要安装的版本号必须小于你查到的版本号！！！ 如下图所示：
我需要安装的是cuda11.8 如果跟我的一样那么请移步到后面百度网盘自取！
具体的安装步骤查阅这位大佬的的博客：https://blog.csdn.net/weixin_45131680/article/details/136314813?utm_medium=distribute.pc_relevant.none-task-blog-2defaultbaidujs_baidulandingword~default-13-136314813-blog-127905428.235v43pc_blog_bottom_relevance_base7&spm=1001.2101.3001.4242.8&utm_relevant_index=16
第二步
安装Anaconda3，这边还是推荐使用这个的，可以很好的管理python版本，就是能够在不影响主体的python版本的前提下能够独立的配置适合项目用的虚拟python版本，不用之后可以随时删掉！安装包放到最后的百度网盘。或者直接在这边下载：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
下载完成之后安装，具体步骤查阅这个博客：https://blog.csdn.net/qq_43674360/article/details/123396415
第三步
这边默认你已经完成了Anaconda3的安装以及环境的配置，接下来需要做的就是配置虚拟环境。
1）进入之后，输入命令：conda create -n pytorch python=3.10.14 python的版本需要和你等下需要安装的pytorch匹配，不然的话你安装不上pytorch会报错。torch1.7以下可以用python3.7/3.6 ，以上的话可以跟我的一样选择高一点的版本。
2）安装结束之后，输入命令conda activate pytorch 进入到刚刚所设置好的虚拟环境。
这个时候我们就要配置自己需要的pytorch环境了。参考官网：https://pytorch.org/get-started/locally/
但是这边建议不要安装最新的，往前面几个版本更稳定一些，所以这个时候你可以访问这个网站：https://pytorch.org/get-started/previous-versions/
搜索选择自己需要的版本安装。
3）我这边安装的是：
pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu118
注意：推荐使用pip安装方式，不用conda的安装方式。主要原因是：conda会根据pytorch版本自己给你升级匹配一些所需的版本号，而pip就是安装你指定的版本来安装，我就这边吃过亏所以值得提醒一下各位。
等待安装完成即可。
第四步
完成这个配置之后则开始安装yolo11,参考GitHub上的步骤：https://docs.ultralytics.com/datasets/segment/
输入命令 pip install ultralytics 即可完成安装yolo11项目环境配置。
按照如下的教程来测试yolo11的性能，具体可以查看上面的连接。
最后
完成了所有的安装，那么就可以慢慢的炼丹了！收工~~
以下是所需要的安装包文件自取：
链接：https://pan.baidu.com/s/1ql0D_768yhceZI_b2LvMXw?pwd=pqm6
提取码：pqm6
原文地址：https://blog.csdn.net/Ppphill_C/article/details/143894279
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！