search

TencentAI4S / tfold

open source code for Tencent tFold
Other
64 stars 9 forks source link

dependency issue in openfold v1.0.1 and pytorch #3

Open atroel01 opened 5 months ago

atroel01 commented 5 months ago

This may help some people avoid a headache... There is a problem with compatibility with newer versions of pytorch and installing openfold v1.0.1. As this is a key dependency on this package i am posting here.

outlined in this issue: https://github.com/facebookresearch/pytorch3d/issues/1657

As Such setup.py within openfold v1.0.1 has to be modified:

extra_cuda_flags = [
    '-std=c++14',
    '-maxrregcount=50',
    '-U__CUDA_NO_HALF_OPERATORS__',
    '-U__CUDA_NO_HALF_CONVERSIONS__',
    '--expt-relaxed-constexpr',
    '--expt-extended-lambda'
]

to

extra_cuda_flags = [
    '-std=c++17',
    '-maxrregcount=50',
    '-U__CUDA_NO_HALF_OPERATORS__',
    '-U__CUDA_NO_HALF_CONVERSIONS__',
    '--expt-relaxed-constexpr',
    '--expt-extended-lambda'
]
Cloud-Rambler commented 5 months ago

Thanks for the tip. It's very helpful

zhanglzu commented 3 months ago

Take my knees

zhanglzu commented 3 months ago

still failed