Commit ·
f525112
1
Parent(s): 501e8b7
debug
Browse files
app.py
CHANGED
|
@@ -6,6 +6,7 @@ import ctypes
|
|
| 6 |
import shlex
|
| 7 |
import torch
|
| 8 |
import argparse
|
|
|
|
| 9 |
print(f'gradio version: {gr.__version__}')
|
| 10 |
|
| 11 |
# Add command line argument parsing
|
|
@@ -76,7 +77,6 @@ def check_gpu():
|
|
| 76 |
print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
|
| 77 |
print("Device count:", torch.cuda.device_count())
|
| 78 |
# test nvdiffrast
|
| 79 |
-
import nvdiffrast.torch as dr
|
| 80 |
dr.RasterizeCudaContext(device="cuda:0")
|
| 81 |
print("nvdiffrast initialized successfully")
|
| 82 |
|
|
|
|
| 6 |
import shlex
|
| 7 |
import torch
|
| 8 |
import argparse
|
| 9 |
+
import nvdiffrast.torch as dr
|
| 10 |
print(f'gradio version: {gr.__version__}')
|
| 11 |
|
| 12 |
# Add command line argument parsing
|
|
|
|
| 77 |
print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
|
| 78 |
print("Device count:", torch.cuda.device_count())
|
| 79 |
# test nvdiffrast
|
|
|
|
| 80 |
dr.RasterizeCudaContext(device="cuda:0")
|
| 81 |
print("nvdiffrast initialized successfully")
|
| 82 |
|