print(torch.cuda.is_available())一直返回false的原因,pytorch需要和cuda有对应版本,具体版本见链接。下载最新pytorch的方法:
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-nightly -c nvidia
一般cuda12.1以上就算比较新了,可以试试最新的pytorch版本。
版本对应链接:pytorch和cuda对应版本