查看cuda版本 |
您所在的位置:网站首页 › 如何查看电脑cuda的版本 › 查看cuda版本 |
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
linux
查看cuda版本
方法一
nvcc --version
或 nvcc -V如果nvcc没有,看方法二。 方法二 去安装目录查看 cat /usr/local/cuda/version.txt 查看cudnn版本 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2如果没有,那么可能没有安装 cuDNN。 如果是为了使用 PyTorch/TensorFlow,在 Linux 服务器上推荐使用 conda 安装,使用 conda 可以很方便安装 PyTorch/TensorFlow 以及对应版本的 CUDA 和 cuDNN。 windows 查看cuda版本在命令行中执行: nvcc --version或者进入 CUDA 的安装目录查看: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA 查看cudnn版本进入 CUDA 的安装目录查看文件 cudnn.h :(注意修改v9.0为自己的版本) C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\cudnn.h具体路径可以去看环境变量。 refer:https://www.cnblogs.com/wuliytTaotao/p/11453265.html |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |