# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

# This Script Assumes Python 3.8.19, CUDA 12.1. Similar package versions might still work but they are not tested.

conda deactivate

# Set environment variables
export ENV_NAME=vfusion3d
export PYTHON_VERSION=3.8.19
export CUDA_VERSION=12.1

# Create a new conda environment and activate it
conda create -n $ENV_NAME python=$PYTHON_VERSION
conda activate $ENV_NAME
conda install pytorch=2.3.0 torchvision==0.18.0 pytorch-cuda=$CUDA_VERSION -c pytorch -c nvidia
pip install transformers
pip install imageio[ffmpeg]
pip install PyMCubes
pip install trimesh
pip install rembg[gpu,cli]
pip install kiui