50 系显卡 (SM120) 科学计算与深度学习环境在 WSL 上的搭建

1

pip install torch torchvision --index-url https://download.pytorch.org/whl/cu130

1

python -c "import torch; print(f\"CUDA available: {'Yes' if torch.cuda.is_available() else 'No'}\"); print(f\"Number of CUDA devices: {torch.cuda.device_count() if torch.cuda.is_available() else 0}\"); print(f\"Device 0: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'No GPU'}\")"

1

pip install openmm[cuda13]

1

python -m openmm.testInstallation