要查看 linux 上的 pytorch 環境,可以依次執行以下步驟:檢查 PyTorch 版本查看已安裝的 PyTorch 包和版本檢查 CUDA 版本(如果已安裝)查看 cuDNN 版本(如果已安裝)檢查 OpenMP 支持檢查 GPU 可用性查看環境變量檢查 python 版本檢查 pip 版本檢查系統信息
如何查看 Linux 上的 PyTorch 環境?
要在 Linux 上查看 PyTorch 環境,請按照以下步驟操作:
1. 檢查 PyTorch 版本(可選):
python -c "import torch; print(torch.__version__)"
2. 查看已安裝的 PyTorch 包和版本:
pip list | grep torch
3. 檢查 CUDA 版本(如果已安裝):
nvcc --version
4. 查看 cuDNN 版本(如果已安裝):
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR
5. 檢查 OpenMP 支持:
python -c "import torch; print(torch.backends.openmp.is_available())"
6. 檢查 GPU 可用性:
nvidia-smi
7. 查看環境變量:
env | grep CUDA env | grep CUDNN
8. 檢查 Python 版本:
python --version
9. 檢查 pip 版本:
pip --version
10. 檢查系統信息:
uname -a cat /etc/os-release
? 版權聲明
文章版權歸作者所有,未經允許請勿轉載。
THE END