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

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

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

官网:https://pytorch.org/get-started/previous-versions/
CUDA Toolkit版本及可用PyTorch对应关系总结(参考官网)

cuda版本确定后,pytorch要依赖此cuda版本,因为pytroch安装跟cuda对应的,比如下图,torch1.11.0只适配cuda10.2、11.3;不适配cuda10.1等其他版本cuda。

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2023年11月7日
下一篇 2023年11月7日

相关推荐