File tree Expand file tree Collapse file tree 1 file changed +7
-7
lines changed Expand file tree Collapse file tree 1 file changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -202,17 +202,17 @@ elif [[ $CUDA_VERSION == "11.8" ]]; then
202
202
export TORCH_NVCC_FLAGS=" -Xfatbin -compress-all --threads 2"
203
203
DEPS_LIST=(
204
204
" /usr/local/cuda/lib64/libcudart.so.11.0"
205
- " /usr/local/cuda/lib64/libnvToolsExt.so.1"
206
- " /usr/local/cuda/lib64/libnvrtc.so.11.2" # this is not a mistake for 11.8, it links to 11.8.89
207
- " /usr/local/cuda/lib64/libnvrtc-builtins.so.11.8"
208
- " $LIBGOMP_PATH "
205
+ " /usr/local/cuda/lib64/libnvToolsExt.so.1"
206
+ " /usr/local/cuda/lib64/libnvrtc.so.11.2" # this is not a mistake for 11.8, it links to 11.8.89
207
+ " /usr/local/cuda/lib64/libnvrtc-builtins.so.11.8"
208
+ " $LIBGOMP_PATH "
209
209
)
210
210
DEPS_SONAME=(
211
211
" libcudart.so.11.0"
212
212
" libnvToolsExt.so.1"
213
213
" libnvrtc.so.11.2"
214
214
" libnvrtc-builtins.so.11.8"
215
- " libgomp.so.1"
215
+ " libgomp.so.1"
216
216
)
217
217
218
218
if [[ -z " $PYTORCH_EXTRA_INSTALL_REQUIREMENTS " ]]; then
@@ -225,10 +225,10 @@ elif [[ $CUDA_VERSION == "11.8" ]]; then
225
225
" /usr/local/cuda/lib64/libcudnn_ops_infer.so.8"
226
226
" /usr/local/cuda/lib64/libcudnn_ops_train.so.8"
227
227
" /usr/local/cuda/lib64/libcudnn.so.8"
228
- " /usr/local/cuda/lib64/libcublas.so.11"
228
+ " /usr/local/cuda/lib64/libcublas.so.11"
229
229
" /usr/local/cuda/lib64/libcublasLt.so.11"
230
230
)
231
- DEPS_SONAME+=(
231
+ DEPS_SONAME+=(
232
232
" libcudnn_adv_infer.so.8"
233
233
" libcudnn_adv_train.so.8"
234
234
" libcudnn_cnn_infer.so.8"
You can’t perform that action at this time.
0 commit comments