关于python pytorch 与CUDA版本相关问题

首先在终端中输入python进入python交互式环境

 import torch
print(torch.__version__)  #注意是双下划线

关于python pytorch 与CUDA版本相关问题_第1张图片

官网:https://pytorch.org/get-started/previous-versions/
CUDA Toolkit版本及可用PyTorch对应关系总结(参考官网)
关于python pytorch 与CUDA版本相关问题_第2张图片

cuda版本确定后,pytorch要依赖此cuda版本,因为pytroch安装跟cuda对应的,比如下图,torch1.11.0只适配cuda10.2、11.3;不适配cuda10.1等其他版本cuda。
关于python pytorch 与CUDA版本相关问题_第3张图片

你可能感兴趣的:(环境配置,python,pytorch,开发语言)