mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-07-31 10:52:50 +08:00
9eeb456e82
Summary: Update all FB license strings to the new format. Reviewed By: patricklabatut Differential Revision: D33403538 fbshipit-source-id: 97a4596c5c888f3c54f44456dc07e718a387a02c
36 lines
1.1 KiB
Bash
Executable File
36 lines
1.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD-style license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
if [[ "$OSTYPE" == "msys" ]]; then
|
|
CUDA_DIR="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v$1"
|
|
else
|
|
CUDA_DIR="/usr/local/cuda-$1"
|
|
fi
|
|
|
|
if ! ls "$CUDA_DIR"
|
|
then
|
|
echo "folder $CUDA_DIR not found to switch"
|
|
fi
|
|
|
|
echo "Switching symlink to $CUDA_DIR"
|
|
mkdir -p /usr/local
|
|
rm -fr /usr/local/cuda
|
|
ln -s "$CUDA_DIR" /usr/local/cuda
|
|
|
|
if [[ "$OSTYPE" == "msys" ]]; then
|
|
export CUDA_VERSION=`ls /usr/local/cuda/bin/cudart64*.dll | head -1 | tr '._' ' ' | cut -d ' ' -f2`
|
|
export CUDNN_VERSION=`ls /usr/local/cuda/bin/cudnn64*.dll | head -1 | tr '._' ' ' | cut -d ' ' -f2`
|
|
else
|
|
export CUDA_VERSION=$(ls /usr/local/cuda/lib64/libcudart.so.*|sort|tac | head -1 | rev | cut -d"." -f -3 | rev)
|
|
export CUDNN_VERSION=$(ls /usr/local/cuda/lib64/libcudnn.so.*|sort|tac | head -1 | rev | cut -d"." -f -3 | rev)
|
|
fi
|
|
|
|
ls -alh /usr/local/cuda
|
|
|
|
echo "CUDA_VERSION=$CUDA_VERSION"
|
|
echo "CUDNN_VERSION=$CUDNN_VERSION"
|