上个月入的5070Ti本来想美美跑代码了,但是发现很多package没有更新到适配cu12.8的版本(.........持续更新一个踩坑记录
1. 装torch的时候会报这个错:
NVIDIA GeForce RTX 5070 Ti with CUDA capability sm_120 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_70 sm_75 sm_80 sm_86 sm_90.
解决方式是只能装pytorch官网的适配cu128的nightly版本:
(哈哈 之前我确实是这样装成功的哈 但是4月14号配另一个环境的时候突然报错了说No matching distribution found for torch也不知道是怎么回事儿
2. 装xformers
xformers目前是完全没有提供cu128对应的包的好像,最后我按照github上贵人的指导进行了重新编译,可以参考:Fixed RTX 5090 build issue by maludwig · Pull Request #1235 · facebookresearch/xformers