torch.cuda.is_available()结果为false
查看服务器cuda版本,nvcc -V 查看pytorch版本对应问题
import torch
print(torch.version)
1.5.1
torch.cuda.is_available()
https://pytorch.org/get-started/previous-versions/
torch.cuda.is_available()结果为false
查看服务器cuda版本,nvcc -V 查看pytorch版本对应问题
import torch
print(torch.version)
1.5.1
torch.cuda.is_available()
https://pytorch.org/get-started/previous-versions/