UnetPointDetection/torch_check.py at 44c43e60007e757fc5be9c0366d283648d0d173d

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))