验证 Anaconda 和 Mamba 安装的 PyTorch 和 CUDA 是否正确
标签
AI
AI/硬件/GPU/NVIDIA
AI/硬件/GPU/NVIDIA/GPU
开发/Python/PyTorch
AI/算法/CUDA
开发/AI/CUDA
字数
143 字
阅读时间
1 分钟
参考安装清单：
yaml
channels:
- pytorch
- nvidia
- conda-forge
dependencies:
- pytorch
- torchvision
- torchaudio
- pytorch-cuda=12.1
- cuda-nvcc检查 PyTorch
启动 python
shell
pythonshell
>>> import torch
>>> torch.version.cuda
'12.1'
>>> torch.backends.cudnn.version()
8902
>>> torch.cuda.is_available()
True或者撰写 Python 脚本和 Notebook：
python
import sys
import torch
print(sys.prefix)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
print(torch.__version__)
print(torch.cuda.is_available())检查 NVCC
运行 nvcc 命令检查是否正确安装 CUDA 编译器
shell
nvcc --versionshell
nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Fri_Dec_17_18:16:03_PST_2021
Cuda compilation tools, release 11.6, V11.6.55
Build cuda_11.6.r11.6/compiler.30794723_0检查 PyTorch 拓展
shell
>>> import torch.utils.cpp_extension
>>> torch.utils.cpp_extension.CUDA_HOME
絢香猫