前言:最近这三天一直在搭配yolo所需要的环境,其实主要问题就是No module named 'torch'
以及No module named 'cv2'
,其中cv2我是有的单独import cv2
也不会报错,但是命令窗内跑代码就提示找不到了。本来1小时之内就可以完成的,但是期间大大小小的坑我几乎都踩了个遍,浪费了很多时间。
希望看到这篇文章的同学可以避坑,真心祝愿!
安装torch需要CUDA和CUDNN(记得给C盘留大一点空间最好大于4G,期间我由于C盘不够耽误了好长时间T_T)
安装CUDA
首先是安装CUDA如果链接失败就是网址更新了,注意年份,当然也可以自行百度。
注释:以下我所有的下载都是用的迅雷(Chrome的下载是真的慢)。
如果访问该页面较慢,