xgsong
Results
3
comments of
xgsong
> 试试这个，TORCH_CUDA_ARCH_LIST="7.5" pip install bmtrain==0.2.1 没用，一样报错，bmtrain与cuda 12不兼容。
> 问题解决方法： 1、将pytorch包内的 site-packages/torch/version.py 内的version 修改为12.1 2、将pytorch包内的 site-packages/torch/nn/modules/module.py 第2112行 中的「, remove_duplicate=remove_duplicate」删除 硬编码的方法确实有问题。
同问，cuda=12.0 python=3.7 torch=1.13，pip install bmtrain就是报错装不上。