torch无法使用cuda
报错检查:
一开始以为是cuda的版本不对,就开始不断的实验各种版本,后来发现可能是使用清华源下载时自动下载的cpu版本。
检查验证:
查看torch版本的python命令,import torch
print(torch.__version__)
,发现输出的确为cpu版本。
处理方法:
用pytorch官网自带的源下载。
torch无法使用cuda
报错检查:
一开始以为是cuda的版本不对,就开始不断的实验各种版本,后来发现可能是使用清华源下载时自动下载的cpu版本。
检查验证:
查看torch版本的python命令,import torch
print(torch.__version__)
,发现输出的确为cpu版本。
处理方法:
用pytorch官网自带的源下载。