We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 14d8b19 commit 4012d26Copy full SHA for 4012d26
README.md
@@ -15,7 +15,7 @@ ml CMake PyTorch CUDA
15
16
- Use `mkdir build` to create your build directory. Change directory into your build folder and compile by running:
17
```bash
18
-cmake -DCUDA_CUDA_LIB=/usr/lib64/libcuda.so -DCMAKE_PREFIX_PATH=`python -c 'import torch;print(torch.utils.cmake_prefix_path)'` ..
+cmake -DCMAKE_PREFIX_PATH=`python -c 'import torch;print(torch.utils.cmake_prefix_path)'` ..
19
cmake --build . --config Release
20
```
21
0 commit comments