yoloV5环境配置及pytorch配置

yoloV5中遇到的坑

使用pip install -r requirement.txt下载依赖,所下载的是torch而非pytorch,且依然能用cpu训练

解决方法

自行下载pytorch和其他依赖


pytorch安装步骤:

  1. 安装cudn显卡驱动(nivida官网下载)
  2. anaconda下载cudatoolkit,cudnn(依赖cudatoolkit),pytorch(依赖cudatoolkit、cuda和python),torchvision和torchaudio(依赖)

例:cudatoolkit _11.3.1 cudnn-8.2.1-cuda11.3_0 pytorch-1.12.0-py3.9_cuda11.3_cudnn8.0 torchvision-0.13.0-py3.9_cu113


** 有时候版本对应但是仍然无法下载,比如上述例子中pytorch-1.12.1-py3.9_cuda11.3_cudnn8.0无法下载,但1.12.0可以下载 **

你可能感兴趣的:(pytorch,深度学习,python)