掘金 人工智能 07月29日 12:21
yoloV5的环境安装
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_guoji1.html

 

本文介绍了使用Python3.8和Anaconda环境,在英伟达RTX4060显卡上安装Pytorch1.8.2的过程,包括安装命令和代码拉取方式。

一. python环境准备

我这边使用的是python3.8和Anaconda管理的环境

二.pytorch安装

    当前的电脑显卡是英伟达RTX4060,电脑是联想拯救者Y7000p, 安装的cuda版本是111,pytorch1.8.2可以正常使用

安装命令: pip install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu111

2. 登录pytorch官网可以看看自己电脑支持什么样版本的

三. 拉取代码安装其他环境

    使用git clone拉取github.com/ultralytics…
    拉下代码后使用

    pip install -r requirements.txt

    安装完成后,python detect.py执行监测命令,如果正常运行,环境就安装成功了

Fish AI Reader

Fish AI Reader

AI辅助创作,多种专业模板,深度分析,高质量内容生成。从观点提取到深度思考,FishAI为您提供全方位的创作支持。新版本引入自定义参数,让您的创作更加个性化和精准。

FishAI

FishAI

鱼阅,AI 时代的下一个智能信息助手,助你摆脱信息焦虑

联系邮箱 441953276@qq.com

相关标签

Python环境 Pytorch安装 CUDA版本
相关文章