Team Ai
Datasetpublic

codekingpro/portable-devtools

sourceHugging Faceupdated 5mo agoView on Hugging Face
1likes16kdownloads
check_gpu.bat10 linesDownload Raw Back to scripts
1@echo off2setlocal3call "%~dp0activate_env.bat"4echo ===================================================5echo           PYTORCH GPU / CUDA CHECKER6echo ===================================================7micromamba run -n pytorch_env python -c "import torch; print('\n[INFO] PyTorch Version:', torch.__version__); print('[INFO] CUDA Available:', torch.cuda.is_available()); print('[INFO] CUDA Version:', torch.version.cuda if torch.cuda.is_available() else 'N/A'); print('[INFO] GPU Name:', torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'None');"8echo ===================================================9pause10 
codekingpro/portable-devtools · Team Ai