1
6
AI人工智能
torchvision版本问题
2021-12-01 16:28:11
205

attachmentId-image.pngwhl里只有torchvision-0.8.0a0+132984f这一个版本吗?怎么安装0.8.1以上版本

全部评论(5)
只看楼主
Z
zhenyi
1楼

在公共目录/public/software/apps/DeepLearning/whl/rocm-4.0.1/路径下有0.9.0和0.10.0的torchvision,可以将对应版本的torchvision中的两个文件夹分别拷贝到自定义环境的site-package中即可。

2021/12/1
A
ac490q7j1h
回复 @zhenyi

感谢回复,我根据帮助文档安装pytorch1.9的环境,torch和torchvision也按照指导安装上了,结果可见下图,但我跑深度学习的时候报错,说是torch和torchvision不匹配错误?查了下torch1.9.0对应torchvision0.10.0,感觉也没什么问题啊,是因为torchvision是0.10.0a0的原因吗?第一轮也跑起来了,但是巨慢,请问该怎么解决呢?

2021/12/2
Z
zhenyi
回复 @zhenyi

我之前用module av,使用pytorch1.8和1.9的时候没有遇到过这样的问题,您可以试一下能不能用。

2021/12/2
A
ac490q7j1h
回复 @zhenyi

您好,我试了1.8和1.9,都还是报同样的错误,是我代码有问题吗,还是配置文件有问题呢?

2021/12/2
全部4条回复
没有更多内容了
--
楼主
Level.2
0 帖子
楼主热帖
暂无数据