Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove hardcoded cuda architecture

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-18 20:44:32 +02:00
parent c1d93bd008
commit ef3a62f399
3 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
CMakeLists.txt
View File

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

2
README.md
View File

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

2
tools/export_model_weights.py
View File

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