pytorch默认安装cpu版本,如何处理
之前电脑上安装过cpu版本的pytorch,现在想安装GPU版本的,从官网复制命令后,下载的还是cpu版本的pytorch
最后发现原因是安装过cpu版本的pytorch后,会给电脑安装一个名字叫做cpuonly的模块,要换成GPU,需要卸载这个模块,我先是
pip uninstall cpuonly,显示卸载成功
但是发现并没有解决问题,后来尝试
conda uninstall pytorch
发现又卸载掉了一个cpuonly的模块,这次成功了
事实上我之前一直用的是pip uninstall pytorch,可能这个命令不会卸载cpuonly