cuda check and better comment

2024-08-22 16:05:56 +02:00
parent c6c7b496c6
commit 483f584e1c
2 changed files with 2 additions and 1 deletions
@@ -1,3 +1,4 @@
+#!/usr/bin/env python
 '''
 little script to check if cuda is available in pytorch and working
 organised as a check that raises an exception if cuda is not available, and data cannot be moved to the gpu