首先在终端中输入python进入python交互式环境
import torch
print(torch.__version__) #注意是双下划线
官网:https://pytorch.org/get-started/previous-versions/
CUDA Toolkit版本及可用PyTorch对应关系总结(参考官网)
文章来源:https://www.toymoban.com/news/detail-739169.html
cuda版本确定后,pytorch要依赖此cuda版本,因为pytroch安装跟cuda对应的,比如下图,torch1.11.0只适配cuda10.2、11.3;不适配cuda10.1等其他版本cuda。
文章来源地址https://www.toymoban.com/news/detail-739169.html
到了这里,关于关于python pytorch 与CUDA版本相关问题的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!