🛠 How to use:

install_nerfstudio_with_colmap_windows.ps1

1. Right-click the file and choose “Run with PowerShell” (or open PowerShell as Administrator).
2. The script will:
   • Install Chocolatey
   • Install Miniconda
   • Set up a Python environment
   • Install PyTorch and tiny-cuda-nn
   • Clone and install Nerfstudio
   • Download and set up COLMAP

# Create a Windows PowerShell script for installing Nerfstudio with COLMAP
windows_script = """
# Nerfstudio + COLMAP Setup Script for Windows (PowerShell)

# === 1. Install Chocolatey if not installed ===
Write-Host "Checking for Chocolatey..."
if (!(Get-Command choco -ErrorAction SilentlyContinue)) {
    Write-Host "Installing Chocolatey..."
    Set-ExecutionPolicy Bypass -Scope Process -Force
    [System.Net.ServicePointManager]::SecurityProtocol = [System.Net.SecurityProtocolType]::Tls12
    iex ((New-Object System.Net.WebClient).DownloadString('<https://chocolatey.org/install.ps1>'))
} else {
    Write-Host "Chocolatey is already installed."
}

# === 2. Install Miniconda ===
Write-Host "Installing Miniconda..."
choco install miniconda3 --yes --params=\\"/AddToPath:1 /RegisterPython:1\\"

# === 3. Refresh environment variables ===
$env:Path = [System.Environment]::GetEnvironmentVariable("Path","Machine")

# === 4. Create Conda Environment ===
Write-Host "Creating Conda environment..."
conda create -y -n nerfstudio-env python=3.8
conda activate nerfstudio-env

# === 5. Install PyTorch with CUDA ===
Write-Host "Installing PyTorch..."
pip install torch torchvision torchaudio --index-url <https://download.pytorch.org/whl/cu116>

# === 6. Install tiny-cuda-nn ===
Write-Host "Installing tiny-cuda-nn..."
$env:TCNN_CUDA_ARCHITECTURES="86"
pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch

# === 7. Clone and Install Nerfstudio ===
Write-Host "Cloning Nerfstudio..."
git clone <https://github.com/nerfstudio-project/nerfstudio.git>
cd nerfstudio
pip install -e .

# === 8. Install COLMAP ===
Write-Host "Downloading COLMAP..."
Invoke-WebRequest -Uri "<https://colmap.github.io/install.html>" -OutFile "COLMAP.zip"
Expand-Archive -Path "COLMAP.zip" -DestinationPath "$env:ProgramFiles\\\\COLMAP"
$colmapPath = "$env:ProgramFiles\\\\COLMAP\\\\COLMAP.bat"
if (Test-Path $colmapPath) {
    Write-Host "COLMAP installed successfully at $colmapPath"
    [Environment]::SetEnvironmentVariable("Path", $env:Path + ";$env:ProgramFiles\\\\COLMAP", "Machine")
} else {
    Write-Host "COLMAP installation failed."
}

# === 9. Test Installation ===
Write-Host "Testing installation..."
ns-train --help
"""

# Save the script as a .ps1 file
script_path = "/mnt/data/install_nerfstudio_with_colmap_windows.ps1"
with open(script_path, "w") as file:
    file.write(windows_script)

script_path