conda安装GPU版pytorch,但是安装的是cpu的
- 安装命令
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
- 安装结果
- 解决办法一(别的博客建议,这里不适合我)
卸载pytorch-mutex
或者卸载cpuonly
conda uninstall pytorch-mutex
- 建议直接使用官网pip安装
可以看到已经变成cuda版本的pytorch了。