Update start.sh

start.sh

@@ -1,5 +1,11 @@
 #!/bin/bash
 
+# 设置错误处理
+set -e
+
+# 创建Hugging Face配置目录
+mkdir -p ~/.huggingface
+
 # 检查是否存在Hugging Face token
 if [ ! -z "$HUGGINGFACE_TOKEN" ]; then
     echo "Setting up Hugging Face credentials..."
@@ -16,5 +22,27 @@ if [ -d "/workspace/android-project" ]; then
     fi
 fi
 
-# 保持容器运行
+# 检查CUDA可用性
+python3 -c "import torch; print('CUDA available:', torch.cuda.is_available())"
+
+# 创建示例Python脚本
+cat > /workspace/test_environment.py << EOL
+from transformers import pipeline
+import torch
+
+print("PyTorch version:", torch.__version__)
+print("CUDA available:", torch.cuda.is_available())
+if torch.cuda.is_available():
+    print("CUDA device:", torch.cuda.get_device_name(0))
+
+# 测试Hugging Face pipeline
+classifier = pipeline("sentiment-analysis")
+result = classifier("Hello, world!")
+print("Test pipeline result:", result)
+EOL
+
+echo "Environment setup complete!"
+echo "You can test the environment by running: python3 /workspace/test_environment.py"
+
+# 执行传入的命令或启动bash
 exec "$@"