widows安装cuda时,cuda版本是11.2,但是官网推荐的torch版本没有11.2的。
可以向下兼容,安装cudatoolkit 11.1 、pytorch 1.8.0 torchvision 0.9。
安装后我有cuda无法调用问题,找原因发现安装的python为3.9,版本过高,换成3.6,成功解决。
widows安装cuda时,cuda版本是11.2,但是官网推荐的torch版本没有11.2的。
可以向下兼容,安装cudatoolkit 11.1 、pytorch 1.8.0 torchvision 0.9。
安装后我有cuda无法调用问题,找原因发现安装的python为3.9,版本过高,换成3.6,成功解决。