之前一直困惑为什么设置好的cuda+cudnn+pytorch只能让一个虚拟环境使用GPU进行训练。
今天终于发现,原来其他虚拟环境的pytorch安装的是cpu版本,并且cuda以及cudnn是不需要重复安装的,只需要卸载CPU版本的pytorch,然后安装对应版本的pytorch即可。
原本在py311这个虚拟环境中可以正常使用
但是在YOLOV5这个环境中无法使用,可以看到pytorch的版本是CPU版本的...
输入下面两行命令,卸载pytorch
pip uninstall torch
pip uninstall torchvision
在官网找到对应版本的pytorch
安装成功。