所讨论的环境是 anaconda ,pytorch 环境
问题:
当项目需要使用GPU的时候,但是没有下载GPU版本的torch和torchvision,下载成了CPU版本的。
在环境中进入python命令行。
如果是这样子,那确定是安装的是CPU版本的。
解决方案:
我的方法:
重新建立一个虚拟环境(因为在原来的基础上不会改,担心改错)。
使用pip安装GPU版本的支持CUDA 11.7的torch和torchvision,并且版本分别为torch==2.0.1和torchvision==0.15.2。具体可以根据需要的版本进行修改。
pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 -f https://download.pytorch.org/whl/torch_stable.html
安装完成之后,检查是否可以正常使用:
这样就完成了,可以使用GPU来加速项目。