bilibili / ailab

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

该升级版本了 it's time to upgrade torch and cuda

shendizhiling opened this issue · comments

cuda走到了12.0,torch走到了2.0
这俩才能完全发挥当前40系卡的能力 但是pytorch1.13不支持cuda1.7以上的版本,4090的最低新架构环境是cuda11.8
3090ti和4090的速度按原始版本还拼不过3090