安装PyTorch及环境配置(应用于Python上的YOLO)

这个基本都是Bilibili网站里面叫“小手丫子”up的视频教程,此前自己需要装了好几次又卸载了好几次,现在根据视频教学整理出来自己所理解的文档。

注意事项

1.安装的pycharm版本和anaconda版本无要求。

2.运行pycharm尽量以管理员身份运行。

3.Cuda是独立的,与显卡驱动无关,与GeForce Experience版本无关。

尽量不要安装Cuda11.8以上版本,用不了TRT。

4.cuda卸载,有关NV的,只保留GFE、图形驱动和Physx,其他的都可以卸载。

安装步骤

安装包链接放在文章最后。

1.安装pycharm

右键点击pycharm的应用程序文件管理员身份打开,开始安装。本文下载链接安装过程中,勾选左边前三个选项,安装完成之后,点击Crack文件的最后一个激活文件,弹框选择确定,等待激活完成。

2.安装anaconda(一定不要装在机械硬盘)。

右键点击Anaconda3的应用程序文件管理员身份打开,开始安装。这个过程如果安装包正常,一般比较顺利也挺快的,如果有安装过程卡顿过久的问题可以参考:

http://t.csdnimg.cn/chprJ

3.安装cuda

右键点击cuda_11.6的应用程序文件管理员身份打开,开始安装。CUDA setup产生的路径为临时文件夹,安装完成会自动删除,可以不用修改。

NVIDIA CUDA安装时,选择自定义安装选项,除了cuda本体组件需要勾选,其他取消勾选。

选择安装位置时,需要记住安装路径,下面需要用到该安装路径。

把cudnn解压后的四个文件复制到cuda的安装路径。

Cuda安装结束。

4.配置环境变量

设置>系统>关于>高级系统设置>环境变量>系统变量>Path>新建;

把cuda安装路径的五个路径:

CUDA_11.6\bin

CUDA_11.6\libnvvp

CUDA_11.6\lib

CUDA_11.6\include

CUDA_11.6\extras\CUPTI\lib64

添加进去,确定。

5.把存有一系列终端指令的.txt和.whl文件放到指定路径

把网盘下载的4个文件:Requirements.txt和torch开头的3个.WHL文件

粘贴到电脑用户的用户名文件夹下。路径在C:\用户\用户名\,下面终端安装直接从该路径选择修改文件配置。

6.安装PyTorch

点击电脑左侧的win, 最近添加, 打开Anaconda Prompt(Anaconda3)的cmd。

(1)先创建一个虚拟环境,在(base) C的用户文件夹下输入命令

conda create --name pytorch python=3.8.15

虚拟环境名字pytorch可根据自己需求修改。

Python建议使用3.8版本,3.8以上版本可能会有问题。

虚拟环境安装好的路径可能会在

users\name\.conda\envs\pytorch

或者C:\ProgramData\Anaconda3\envs\pytorch

(2)切换到创建好的虚拟环境pytorch(创建好环境之后进入虚拟环境也可以Win+R进入终端), 输入

conda activate pytorch

在pytorch下,要保证之前4个文件已经复制到用户名文件夹下,输入

pip install -r requirements.txt -i https://pypi.mirrors.ustc.cdu.cn/simple

安装这个txt文件里面的一些依赖包,使用国内源安装,速度很快。

如果出现报错:

ERROR:Could not find a version that satisfies the requirement absl-py==1.3.0 (from versions: none)

ERROR: No matching distribution found for absl-py==1.3.0

可以先在把pytorch的虚拟环境关闭,打开Anaconda Prompt的终端,pip list查看是否有pip,没有则安装,有则需要升级,输入:

pip install --upgrade pip

再切换进去创建的python虚拟环境,继续输入上面的指令安装txt文件里面的一些依赖包。

未报错则直接安装,完成后安装pytorch,输入

pip install torch-1.13.1+cu116-cp38-cp38-win_amd64.whl

完成后,再输入

pip install torchaudio-0.13.1+cu116-cp38-cp38-win_amd64.whl

完成后,再输入

pip install torchvision-0.14.1+cu116-cp38-cp38-win_amd64.whl

都安装完成后,输入

pip list

检查安装版本是否都在。

(3)确认全部安装完成之后,就可以把用户名文件夹下之前复制过来的4个文件(一个txt和三个whl文件)删除。

(4)把yolo源码放入一个文件夹,右键使用Pycharm打开项目。

(5)在pycharm界面添加python解释器,

右下角解释器设置>python解释器>点击右边下箭头>全部显示>点击+>现有环境>选择刚才创建的虚拟环境路径

C:\User\name\conda\envs\pytorch\python.exe

确定。

此时YOLOv5就可以在运行起来了。

各个安装包下载链接

pychram2022链接:https://pan.baidu.com/s/1IsPUFG_-HPbo6Jh77-ilgg 

python3.8.0链接:https://pan.baidu.com/s/1KmPYoF3h80EB56r0oqD2iw 

Anaconda3_2023链接:https://pan.baidu.com/s/1iDyqFli9slb81PiRu5MsfQ 

cuda11.6和cudnn链接:https://pan.baidu.com/s/1KGT9hwrOd6RP9vYr1li-dQ 

安装pytorch的依赖包指令文件链接:https://pan.baidu.com/s/10Yw5mf2AA1lrfpyQHx9DEA 

YOLOv5_7.0源码文件链接:https://pan.baidu.com/s/1ph3QRq5fhlp28XN_fwhNRA 

提取码:lt24

相关推荐

  1. yarn安装配置

    2024-01-10 23:02:03       40 阅读
  2. Cuda环境Pytorch环境安装

    2024-01-10 23:02:03       25 阅读
  3. python安装环境配置相关问题记录

    2024-01-10 23:02:03       30 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-01-10 23:02:03       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-01-10 23:02:03       106 阅读
  3. 在Django里面运行非项目文件

    2024-01-10 23:02:03       87 阅读
  4. Python语言-面向对象

    2024-01-10 23:02:03       96 阅读

热门阅读

  1. Vue项目里实现json对象转formData数据

    2024-01-10 23:02:03       58 阅读
  2. 镜头选型和计算

    2024-01-10 23:02:03       64 阅读
  3. Linux read命令详解

    2024-01-10 23:02:03       51 阅读
  4. 题记(9)--简单密码

    2024-01-10 23:02:03       36 阅读
  5. K8S集群调度(2)

    2024-01-10 23:02:03       60 阅读
  6. C++常见错误(汇总版1)

    2024-01-10 23:02:03       48 阅读
  7. Docker:docker exec命令简介

    2024-01-10 23:02:03       56 阅读
  8. 【docker】常用指令整理

    2024-01-10 23:02:03       45 阅读
  9. 题记(10)--大数加法

    2024-01-10 23:02:03       52 阅读