环境搭建--anaconda环境下运行YoloV5的torch环境配置

文章用于记录,防止以后忘记~

系统环境:Ubuntu16.04

驱动与CUDA版本:NVIDIA-SMI 430.64       Driver Version: 430.64       CUDA Version: 10.1

Python版本:3.8.1

配置前应先下载anaconda

1.创建虚拟环境

conda create -n yolo python=3.8.1

2.激活虚拟环境

conda activate yolo

3.配置CUDA和CUDNN

pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

配置完成后,可在虚拟环境下运行YoloV5的代码了!

 参考自:安装cuda10.1对应的pytorch_sinat_36234054的博客-CSDN博客