ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-13 16:53:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Code relocation

Browse Source
This commit is contained in:
superobk
2024-04-13 00:14:26 +08:00
parent f606db69c7
commit c46c0ff7f2
4 changed files with 2 additions and 27494 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
sample/cuda.py Normal file
View File

@ -0,0 +1,5 @@
import torch
print("Torch version:",torch.__version__)
print("Is CUDA enabled?",torch.cuda.is_available())