Scruff AI
27 lines
911 B
Bash
27 lines
911 B
Bash
#!/bin/bash
|
|
export PATH=/usr/local/cuda-12.6/bin:$PATH
|
|
export LD_LIBRARY_PATH=/usr/local/cuda-12.6/lib64:$LD_LIBRARY_PATH
|
|
|
|
echo "=== VERIFY INSTALL ==="
|
|
echo -n "nvcc: "
|
|
nvcc --version 2>&1 | grep "release" || echo "NOT FOUND"
|
|
echo ""
|
|
echo -n "zmq: "
|
|
dpkg -l libzmq3-dev 2>/dev/null | grep "^ii" | awk '{print $2, $3}' || echo "NOT FOUND"
|
|
echo -n "json-c: "
|
|
dpkg -l libjson-c-dev 2>/dev/null | grep "^ii" | awk '{print $2, $3}' || echo "NOT FOUND"
|
|
echo -n "nvml-dev: "
|
|
dpkg -l cuda-nvml-dev-12-6 2>/dev/null | grep "^ii" | awk '{print $2, $3}' || echo "NOT FOUND"
|
|
|
|
# Add to bashrc if not already
|
|
if ! grep -q "cuda-12" ~/.bashrc 2>/dev/null; then
|
|
echo 'export PATH=/usr/local/cuda-12.6/bin:$PATH' >> ~/.bashrc
|
|
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.6/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
|
|
echo "Added CUDA to .bashrc"
|
|
else
|
|
echo "CUDA already in .bashrc"
|
|
fi
|
|
|
|
echo ""
|
|
echo "=== READY ==="
|