cc_staff
653
edits
PyTorch: Difference between revisions
Jump to navigation
Jump to search
m
forgot to "make"
m (also wrap the cmake command for StdEnv/2020 to make it more readable) |
m (forgot to "make") |
||
| Line 2,219: | Line 2,219: | ||
-DCMAKE_EXE_LINKER_FLAGS=-Wl,-rpath=$VIRTUAL_ENV/lib/python3.11/site-packages/torch/lib,-L$EBROOTCUDA/extras/CUPTI/lib64 \ | -DCMAKE_EXE_LINKER_FLAGS=-Wl,-rpath=$VIRTUAL_ENV/lib/python3.11/site-packages/torch/lib,-L$EBROOTCUDA/extras/CUPTI/lib64 \ | ||
-DCMAKE_SKIP_RPATH=ON -DTORCH_CUDA_ARCH_LIST="6.0;7.0;7.5;8.0;9.0" | -DCMAKE_SKIP_RPATH=ON -DTORCH_CUDA_ARCH_LIST="6.0;7.0;7.5;8.0;9.0" | ||
cmake --build build | |||
</tab> | </tab> | ||
<tab name="StdEnv/2020"> | <tab name="StdEnv/2020"> | ||