forked from yilundu/ired_code_release
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathinstall.sh
More file actions
executable file
·23 lines (18 loc) · 947 Bytes
/
Copy pathinstall.sh
File metadata and controls
executable file
·23 lines (18 loc) · 947 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#!/bin/bash
# Installation script for IRED project
# GPU: NVIDIA GeForce RTX 3090, CUDA 12.9
echo "Installing IRED dependencies for CUDA 12.9..."
echo "GPU: NVIDIA GeForce RTX 3090"
echo ""
# Install PyTorch with CUDA 12.1 support (compatible with CUDA 12.9)
echo "Installing PyTorch with CUDA 12.1 support..."
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu121
# Install remaining dependencies
echo ""
echo "Installing other dependencies..."
pip3 install einops>=0.6.0 numpy>=1.21.0 scipy>=1.7.0 pandas>=1.3.0 matplotlib>=3.4.0 tqdm>=4.62.0 Pillow>=8.0.0
echo ""
echo "Installation complete!"
echo ""
echo "Testing PyTorch CUDA availability..."
python3 -c "import torch; print(f'PyTorch version: {torch.__version__}'); print(f'CUDA available: {torch.cuda.is_available()}'); print(f'CUDA version: {torch.version.cuda}'); print(f'GPU: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else \"N/A\"}')"