pip install pytorch-directml
pip install tensorflow-directml
检测是否可用
torch.cuda.is_available()
torch.cuda.device_count()
torch.cuda.get_device_name(0)
torch.__version__
pip install pytorch-directml
pip install tensorflow-directml
检测是否可用
torch.cuda.is_available()
torch.cuda.device_count()
torch.cuda.get_device_name(0)
torch.__version__