26 lines
901 B
Bash
26 lines
901 B
Bash
export CUDA_HOME=/usr/local/cuda-11.8
|
|
export PATH=/usr/local/cuda-11.8/bin:$PATH
|
|
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
|
|
export DEBIAN_FRONTEND=noninteractive
|
|
export TORCH_CUDA_ARCH_LIST="8.0;8.9;9.0"
|
|
export TCNN_CUDA_ARCHITECTURES=80,89,90
|
|
|
|
source /opt/conda/etc/profile.d/conda.sh
|
|
conda activate embodiedgen
|
|
chmod -R 777 /EmbodiedGen/apps/sessions
|
|
DATE_STR=$(date +"%Y-%m-%d_%H-%M-%S")
|
|
|
|
LOG_FILE="${DATE_STR}.log"
|
|
exec > >(tee -a "/EmbodiedGen/logs/${LOG_FILE}") 2>&1
|
|
|
|
export no_proxy="localhost,127.0.0.1,192.168.48.210,120.48.161.22"
|
|
export ENDPOINT="https://llmproxy.d-robotics.cc/v1"
|
|
export API_KEY="sk-B8urDShf4TLeruwI3dB8286485Aa4984A722E945F566EfF4"
|
|
export MODEL_NAME="azure/gpt-4.1"
|
|
|
|
export http_proxy=http://192.168.16.68:18000 https_proxy=http://192.168.16.68:18000
|
|
if [ -z "$GPUS" ]; then
|
|
GPUS=0
|
|
fi
|
|
CUDA_VISIBLE_DEVICES=$GPUS python image_to_3d_service.py
|