训练界面没有GPU选项

[cdbmfrcjj]

Search before asking

• I have searched the X-AnyLabeling Docs and issues and found no similar questions.

Question

我安装的GPU版本 但是 训练界面 没有GPU选项

Additional

No response

[CVHub520]

Here are the devices that were detected:

X-AnyLabeling/anylabeling/services/auto_training/ultralytics/config.py

Lines 137 to 144 in 58b1f25

	IS_TORCH_AVAILABLE = is_torch_available()
	IS_CUDA_AVAILABLE = is_cuda_available()
	IS_MPS_AVAILABLE = is_mps_available()
	DEVICE_OPTIONS = (
	(["cuda"] if IS_CUDA_AVAILABLE else [])
	+ (["mps"] if IS_MPS_AVAILABLE else [])
	+ ["cpu"]
	)

[goldenzjy2023]

按文档教程安装应该是没有开启cuda的，你需要根据gpu的cuda版本和python版本，安装对应的torch；在python命令里用
import torch
print(torch.cuda.is_available())
如果打印出true才说明开启了cpu。如果是false，那你环境还有缺少或版本不一直，如cuda版本、flash版本、torchvision、torchaudio