conda安装pytorch环境（与YOLO v5匹配）（Linux）

我在学习中……

已于 2022-07-28 16:50:11 修改

阅读量602

点赞数 1

分类专栏：环境配置文章标签： python 深度学习 pytorch 目标检测

于 2022-07-28 16:37:41 首次发布

本文链接：https://blog.csdn.net/qq_47676395/article/details/126037516

版权

2 篇文章 1 订阅

订阅专栏

该博客详细介绍了如何通过conda创建环境并安装PyTorch、CUDA、CuDNN，以及如何从清华大学开源软件镜像站下载PyTorch。之后，博主指导读者克隆Yolov5项目，安装依赖包，为进行目标检测做好准备。参考链接提供了Yolov5的官方文档，供进一步学习。

摘要由CSDN通过智能技术生成

一、安装pytorch（yolo v5匹配）

conda create -n <name> python==3.8

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

完整的代码如下：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

conda install cudatoolkit=11.3.1

conda install cudn=8.2.1

克隆yolov5：

https://docs.ultralytics.com/tutorials/train-custom-datasets/
git clone https://github.com/ultralytics/yolov5

cd yolov5

下载依赖包：

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

关注

专栏目录