Linux环境下conda环境中无法配置CUDA版torch解决方法

出现的问题

Windows环境下使用conda配置CUDA版torch顺风顺水，结果到了Linux环境下，按照之前一模一样的配置方法，却一直在print(torch.cuda.is_available())打印出false，打印torch版本print(torch.__version__)也显示2.0.1(下的版本是2.0.1+cu118)
是按照pytorch官网的下载方法下载的conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia,使用conda -list命令查看也显示有此cuda版本的torch库

解决方法

先新建一个conda环境conda create -n xxx python=x.x，再进入此环境conda activate xxx

(xxx)root@root:

然后pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
对torch的cuda版本进行安装，使用print(torch.cuda.is_available())打印出true,print(torch.__version__)显示2.0.1+cu118