一、问题描述
根据官网的指令去安装pytorch,明明指令是GPU版本的,但是安装结果就是CPU版本的
二、解决方法
先进入你的虚拟环境检查一下有没有cpuonly的库
conda list
有就
conda uninstall cpuonly
没有就先安装再卸载
conda install cpuonly
conda uninstall cpuonly
根据官网的指令去安装pytorch,明明指令是GPU版本的,但是安装结果就是CPU版本的
先进入你的虚拟环境检查一下有没有cpuonly的库
conda list
有就
conda uninstall cpuonly
没有就先安装再卸载
conda install cpuonly
conda uninstall cpuonly