如何查看自己的cuda环境,以及其对应版本

  • Post author:
  • Post category:其他



第一步:

cmd打开命令窗口,输入conda-env list 查看有几个conda环境


第二步:

选择一个环境进入conda,输入conda activate pytorch


第三步:

进入python环境,输入python


第四部 :

在Python中,查看torch、cuda和cudnn的版本

1. 查看torch版本

>>> import torch

>>> print(torch.__version__)

1.2.0

2.查看cuda版本

>>> print(torch.version.cuda)

10.0

3.查看cudnn版本

>>> print(torch.backends.cudnn.version())

7401

torch torchvision cuda python对应版本

torch、torchvision、cuda 、python对应版本匹配

pip install torch == 1.3.0 torchvision ==0.4.1 默认相应cuda最高版本10.1

参考:https://pytorch.org/get-started/previous-versions/



版权声明:本文为weixin_45819759原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。