Spaces:
Starting
on
L40S
Starting
on
L40S
File size: 657 Bytes
d7e58f0 |
1 2 3 4 5 6 7 8 9 10 11 12 |
#!/usr/bin/env bash
# ------------------------------------------------------------------------------------------------
# Deformable DETR
# Copyright (c) 2020 SenseTime. All Rights Reserved.
# Licensed under the Apache License, Version 2.0 [see LICENSE for details]
# ------------------------------------------------------------------------------------------------
# Modified from https://github.com/chengdazhi/Deformable-Convolution-V2-PyTorch/tree/pytorch_1.0.0
# ------------------------------------------------------------------------------------------------
TORCH_CUDA_ARCH_LIST="8.0" CUDA_HOME='path/to/your/cuda_home' python setup.py build install
|