如何查看pytorch对应的cuda版本

如何查看pytorch对应的cuda版本?

打开终端输入:python

输入命令:import torch

再输入:print(torch.version.cuda)

另外,pytorch安装(conda,pip)等等需要看官网要求:https://pytorch.org/

发布了191 篇原创文章 · 获赞 104 · 访问量 34万+

猜你喜欢

转载自blog.csdn.net/u013925378/article/details/103045065