# pytorch查看版本
python -c "import torch; print(torch.__version__)"
# 查看显存
python -c "import torch; print('Total CUDA Memory:', torch.cuda.get_device_properties(0).total_memory / (1024**3), 'GB', 'Allocated Memory:', torch.cuda.memory_allocated(0) / (1024**3), 'GB')"
# python查看版本 
python -V
# 查看 CUDA 版本
nvcc --version
# 查看支持版本
nvidia-smi