Remove hardcoded cuda architecture

CMakeLists.txt

@@ -19,7 +19,6 @@ set(LIBRARY_SOURCES
     ${LIBRARY_SOURCES}
 )
 
-set(CMAKE_CUDA_ARCHITECTURES 75)
 set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
 
 # Build static library

README.md

@@ -29,7 +29,7 @@ Convolutional Neural Network inference library running on CUDA.
 ```sh
 mkdir build
 cd build
-cmake -S ..
+cmake -S .. -DCMAKE_CUDA_ARCHITECTURES=75  # Replace with you cuda architecture
 make
 ```
 

tools/export_model_weights.py

@@ -25,4 +25,4 @@ def export_model_weights(model: torch.nn.Module, filename):
 
         f.seek(0)
         f.write(struct.pack('q', len(header)))            
-        f.write(header.encode('utf-8'))
+        f.write(header.encode('utf-8'))