import resample2d_cuda ImportError

[wbiao98]

when I tried this : python demo.py
--name=pose_fashion_checkpoints
--model=pose
--attn_layer=2,3
--kernel_size=2=5,3=3
--gpu_id=0
--dataset_mode=fashion
--dataroot=./dataset/fashion
--results_dir=./demo_results/fashion

then met this: File "demo.py", line 1, in from options.val_options import ValOptions File "/home/ubuntu/WB/Global-Flow-Local-Attention/options/val_options.py", line 1, in from .base_options import BaseOptions File "/home/ubuntu/WB/Global-Flow-Local-Attention/options/base_options.py", line 4, in import model File "/home/ubuntu/WB/Global-Flow-Local-Attention/model/init.py", line 4, in from .base_model import BaseModel File "/home/ubuntu/WB/Global-Flow-Local-Attention/model/base_model.py", line 6, in from model.networks import base_function File "/home/ubuntu/WB/Global-Flow-Local-Attention/model/networks/init.py", line 3, in from model.networks.discriminator import * File "/home/ubuntu/WB/Global-Flow-Local-Attention/model/networks/discriminator.py", line 6, in from model.networks.base_function import * File "/home/ubuntu/WB/Global-Flow-Local-Attention/model/networks/base_function.py", line 10, in from model.networks.resample2d_package.resample2d import Resample2d File "/home/ubuntu/WB/Global-Flow-Local-Attention/model/networks/resample2d_package/resample2d.py", line 4, in import resample2d_cuda ImportError: libcudart.so.10.0: cannot open shared object file: No such file or directory so how can i get this resample2d_cuda

[MeaninglessAI]

3. Build pytorch Custom CUDA Extensions

./setup.sh