0.前提:Ubuntu上已经有显卡驱动,终端输入
nvidia-smi
可查到。1.安装anaconda,官网下载Linux版本。
1)找到打开下载文件的位置:打开terminal
2)运行 .sh 文件:
bash Anaconda3-5.2.0-Linux-x86_64.sh
3)一路enter;
4)刷新环境变量
sudo gedit ~/.bashrc
export PATH="/home/xupp/anaconda3/bin:$PATH"
source ~/.bashrc
5)检验
anaconda --version
2.创建虚拟环境,并进入环境
conda create --name yolov5 python=3.8
conda activate yolov5
3.安装cuda11.3(由第一个图可以看出,对应的cuda版本) ,pytorch官网查询指令
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
4.之后安装yolov5例程里的其他包
pip install 包名 -i https://pypi.tuna.tsinghua.edu.cn/simple/
5.配置结束,下载yolov5例程选择配置好的环境即可。
6.最后可以检验cuda是否配置好
import torch
import torchvision
print(torch.__version__)
print(torch.cuda.is_available())
拓展:
生成requirements.txt文件
pip freeze > requirements.txt
安装requirements.txt依赖
pip install -r requirements.txt