File size: 1,545 Bytes
2072243 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | #!/bin/bash
# setup_cuda.sh — FusionNet NVIDIA CUDA setup for Linux / WSL2
# ──────────────────────────────────────────────────────────────
# Requires: NVIDIA driver >= 560 for CUDA 12.8
# Windows users: use scripts\setup_cuda.ps1 instead (native PowerShell)
# ──────────────────────────────────────────────────────────────
set -e
echo "Setting up NVIDIA CUDA environment for FusionNet..."
# Requires driver >= 560 for CUDA 12.8 (Blackwell RTX 50xx, Ada RTX 40xx, Ampere RTX 30xx).
# Install PyTorch cu128 — the first stable wheel set supporting Blackwell (sm_100).
echo "Installing PyTorch cu128..."
pip install torch==2.11.0+cu128 torchvision==0.26.0+cu128 torchaudio==2.11.0+cu128 \
--index-url https://download.pytorch.org/whl/cu128
# Install remaining dependencies (torch is excluded from requirements.txt)
pip install -r ../requirements.txt
echo ""
echo "Setup complete. Verifying environment..."
python -c "
import os, sys
if sys.platform == 'win32':
os.add_dll_directory(r'C:\Windows\System32')
import torch
print(f'PyTorch : {torch.__version__}')
print(f'CUDA : {torch.cuda.is_available()}')
if torch.cuda.is_available():
p = torch.cuda.get_device_properties(0)
print(f'GPU : {p.name} ({p.total_memory / 1024**3:.1f} GB)')
"
|