elungky commited on
Commit
6607715
·
1 Parent(s): ea133b8

Added COPY . /app to Dockerfile to include project files

Browse files
Files changed (1) hide show
  1. start.sh +3 -3
start.sh CHANGED
@@ -1,9 +1,9 @@
1
  #!/bin/bash
2
 
3
  # Set environment variables for a single GPU on Hugging Face Spaces
4
- export CUDA_VISIBLE_DEVICES="0" # Hugging Face maps the visible GPU to device 0
5
- export CUDA_HOME="/usr/local/cuda" # Common path for CUDA toolkit in Docker images
6
- export PYTHONPATH="/app" # Assuming /app is your WORKDIR in the base image where your code is
7
 
8
  echo "Starting GEN3C application on A100 Large 80GB GPU..."
9
 
 
1
  #!/bin/bash
2
 
3
  # Set environment variables for a single GPU on Hugging Face Spaces
4
+ export CUDA_VISIBLE_DEVICES="0"
5
+ export CUDA_HOME="/usr/local/cuda"
6
+ export PYTHONPATH="/app"
7
 
8
  echo "Starting GEN3C application on A100 Large 80GB GPU..."
9