Make PyTorch bindings installation more ergonomic (no static CUDA)

NVlabs · Apr 19, 2023 · 52e560d · 52e560d
1 parent 922bb97
commit 52e560d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -1,6 +1,8 @@
 .DS_Store
 *.egg-info
 __pycache__
+*.o
+/*.jpg
 build*
 dist
 /.vscode
diff --git a/bindings/torch/setup.py b/bindings/torch/setup.py
@@ -165,7 +165,7 @@ def make_extension(compute_capability):
 			"%s/dependencies/fmt/include" % root_dir,
 		],
 		extra_compile_args={"cxx": cflags, "nvcc": nvcc_flags},
-		libraries=["cuda", "cudadevrt", "cudart_static"],
+		libraries=["cuda"],
 	)
 	return ext
 

diff --git a/samples/mlp_learning_an_image_pytorch.py b/samples/mlp_learning_an_image_pytorch.py