1  #!/bin/bash
 2  
 3  # 设置环境变量抑制TensorFlow AVX警告
 4  export TF_CPP_MIN_LOG_LEVEL=2
 5  export TF_ENABLE_ONEDNN_OPTS=0
 6  
 7  # 设置其他有用的环境变量
 8  export CUDA_VISIBLE_DEVICES=0  # 如果有多个GPU，可以指定使用哪个
 9  export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:128
10  
11  echo "环境变量设置完成:"
12  echo "TF_CPP_MIN_LOG_LEVEL=$TF_CPP_MIN_LOG_LEVEL"
13  echo "TF_ENABLE_ONEDNN_OPTS=$TF_ENABLE_ONEDNN_OPTS"
14  echo "CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES"
15  echo "PYTORCH_CUDA_ALLOC_CONF=$PYTORCH_CUDA_ALLOC_CONF"