使用conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
安装pytorch,执行mport torch; torch.cuda.is_available()
总是显示false,是因为安装了cpu版本的pytorch
参考这篇文章的解释,conda在源里找不到gpu版本的会给你安装一个cpu版本的。
但是上面那篇文章安装的pytorch的版本太低了
所以推荐别用conda安装,而是用pip安装: