参考资料:http://t.csdnimg.cn/kjsCr
一、创建虚拟环境
conda create -n yolo5.7-GPU
激活环境:conda activate yolo5.7-gpu(退出:conda deactivate)
二、安装CUDA
1、查看GPU版本
nvidia-smi
2、安装
参考资料:http://t.csdnimg.cn/0vrdq
三、安装cudnn
参考资料:深度学习之CUDA+CUDNN详细安装教程 - 知乎
不过看这个评论,现在下载pytorch,自带cuda和cudnn OMG.
四、验证是否安装成功
参考链接:http://t.csdnimg.cn/p71Zj
五、安装pytorch