Pytorch无法调用cuda
测试代码
import torch
print(torch.__version__)
print(torch.cuda.is_available())
正确输出
重点!!!!
请一定安装官网安装torch包
https://pytorch.org/get-started/previous-versions/
测试代码
import torch
print(torch.__version__)
print(torch.cuda.is_available())
正确输出
重点!!!!
请一定安装官网安装torch包
https://pytorch.org/get-started/previous-versions/