You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
6 lines
242 B
Python
6 lines
242 B
Python
5 months ago
|
import torch
|
||
|
|
|
||
|
|
if torch.cuda.is_available():
|
||
|
|
print('CUDA 可用，如果实际使用仍提示 cuda 相关错误，请尝试升级显卡驱动\n将 set.init 中 devtype=cpu改为devtype=cuda')
|
||
|
|
else:
|
||
|
|
print("当前计算机CUDA不可用")
|