如何查看cuda版本和pytorch版本
时间: 2023-12-23 18:05:37 浏览: 333
pytorch 查看cuda 版本方式
5星 · 资源好评率100%
要查看CUDA版本和PyTorch版本,可以按照以下步骤进行操作:
1. 查看CUDA版本:可以在命令行中输入“nvcc --version”来查看CUDA版本。
2. 查看PyTorch版本:可以在Python环境中导入PyTorch并打印版本号。具体步骤如下:
a. 查看所有环境信息:conda info --envs
b. 找到安装有PyTorch的环境并激活:source activate torch_env
c. 进入Python运行环境:python
d. 导入PyTorch并打印版本号:import torch 换行 print(torch.__version__)
另外,如果需要查看NVIDIA显卡的相关信息,可以按照引用中的方法添加环境变量并使用命令行工具NVSMI来查看。
阅读全文