Skip to content
Snippets Groups Projects
Commit 1a8a58e5 authored by Matthew Lewis's avatar Matthew Lewis
Browse files

added CUDA check

parent 4940569c
Branches main
No related merge requests found
......@@ -2,7 +2,7 @@
"name": "adpkd-segmentation",
"label": "ADPKD segmentation",
"description": "ADPKD-segmentation for determing Total Kidney Volume - TKV (https://github.com/aksg87/adpkd-segmentation-pytorch) - CPU only",
"version": "0.1_gpu",
"version": "0.2_gpu",
"author": "Radiology @ UTSW",
"maintainer": "Matthew Lewis <Matthew.Lewis@UTSouthwestern.edu>",
"cite": "",
......@@ -23,7 +23,7 @@
},
"gear-builder": {
"category": "analysis",
"image": "ai/adpkd-segmentation:0.1_gpu"
"image": "ai/adpkd-segmentation:0.2_gpu"
}
},
"inputs": {
......
......@@ -5,7 +5,8 @@ import os
import logging
import sys
import shutil
import glob
import glob
import torch
from flywheel_gear_toolkit import GearToolkitContext
import flywheel
......@@ -13,6 +14,7 @@ import flywheel
log = logging.getLogger(__name__)
def main(context):
print('Is CUDA available for PyTorch? ',torch.cuda.is_available(), flush=True)
# print(dir(context))
print("config_json:")
print(context.config_json)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment