在 Ubuntu 上查看已安装的 PyTorch 版本,常用方法如下:
python3 -c "import torch; print(torch.__version__)"
或进入 Python 交互环境:
python3
>>> import torch
>>> print(torch.__version__)
如果你用的是 conda 环境,先激活环境再执行:
conda activate your_env_name
python -c "import torch; print(torch.__version__)"
python3 -c "import torch; print(torch.version.cuda)"
None:CPU 版本11.8:CUDA 版本还可以看是否能用 GPU:
python3 -c "import torch; print(torch.cuda.is_available())"
pip3 show torch
或
pip3 list | grep torch
如果是 conda 安装的:
conda list | grep pytorch
pip3 show torch | grep Location
如果你愿意,可以告诉我: