Ubuntu16.04配置YOLOv5环境

一、安装显卡驱动

为了安装显卡驱动真的经历了太多。。。刚开始是在Ubuntu20.04上安装显卡驱动,出现了很多问题,一一解决后,又卡在 tty1 进不去图形界面了,遂卸载重装了Ubuntu16.04。

装完Ubuntu16.04后发现上不了网,原因没有网卡驱动,网卡驱动安装教程

安装显卡驱动有几个小问题要注意:

1、tty1界面用户登录的时候不能用小键盘,一般都有问题,可以用字母上方的键盘

2、驱动不要下到中文路径

 3、tty1界面乱码解决办法

二、安装CUDA10.02以及cudnn

 有一个大无语事件,最后安装好测试输入命令nvcc -V,这个 V 必须是大写的,我用小写的结果输出 nvcc fatal ,我还捣鼓了半天以为没安装好。

三、安装anaconda3 

anaconda安装比较简单,参照这篇文章即可,安装好后输入以下指令测试

conda --version

如果出现 conda:未找到命令,可参考这篇博客

四、anaconda 创建虚拟环境

yolov5 要求 python 版本大于等于 3.7

source activate  // 进入anaconda
conda create -n name python=3.7  // name 是创建的环境的名字  我取的是yolov5
conda activate yolov5  // 进入自己创建的环境 

安装 pyTouch ,进入官网获取对应版本的安装命令,执行安装命令安装。

Ubuntu16.04配置YOLOv5环境_第1张图片

 安装完成之后,在终端输入python,随后输入以下代码:

import torch
print(torch.cuda.is_available())

输出 true 表示安装成功。

你可能感兴趣的:(Linux,linux)