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