TVConv icon indicating copy to clipboard operation
TVConv copied to clipboard

版本问题

Open zi1zi opened this issue 2 years ago • 4 comments

您好,pytorch必须是1.71吗,其他版本可以吗,我的cuda是11.1的

zi1zi avatar Oct 10 '22 11:10 zi1zi

你好,pytorch版本不一定必须是1.7.1,只需要和pytorchlighting兼容即可。我们采用的pytorch-lightning版本为1.4.6,如果升到最新版本可能会有api不一致的问题,需要相应更改api。

JierunChen avatar Oct 10 '22 12:10 JierunChen

谢谢您的及时回复,因为我的cuda是11.1,所以下载的是pytorch1.80,torchvision==0.9.0,pytorchlighting=1.46,结果出现了一下问题,网上的一些答复是pytorch和torchvision版本不对应,但是我是从官网下载的这是对应的呀 image

zi1zi avatar Oct 10 '22 12:10 zi1zi

谢谢您的及时回复,因为我的cuda是11.1,所以下载的是pytorch1.80,torchvision==0.9.0,pytorchlighting=1.46,结果出现了一下问题,网上的一些答复是pytorch和torchvision版本不对应,但是我是从官网下载的这是对应的呀 image

zi1zi avatar Oct 10 '22 12:10 zi1zi

所以请问下到底可以用什么版本的pytorch和torchvision啊

likeown2003 avatar May 04 '24 19:05 likeown2003