UnetPointDetection/torch_check.py at 333f85b2c936c52ac2a85ce50df50d9ef5366758

Fork: 0

nakaguchi / UnetPointDetection

Find file

Newer

Older

UnetPointDetection / torch_check.py

T.Nakaguchi (Toshiya Nakaguchi) 12 days ago 208 bytes 針先端検出の学習と推論基本実装

Raw Blame History

import torch

# Pytorchのバージョン
print(torch.__version__)

# GPUが利用可能であるか否か
print(torch.cuda.is_available())

# 利用可能なGPUの名前
print(torch.cuda.get_device_name(0))