Skip to content

Commit

Permalink
feat: update environments for speed test
Browse files Browse the repository at this point in the history
  • Loading branch information
khwengXU committed Jul 3, 2022
1 parent 9894375 commit 2e1c276
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/Test_speed.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ Download the models you want to test from the latest release.

Refer to README, install packages corresponding to CUDA, CUDNN and TensorRT version.

Here, we use Torch1.8.0 inference on V100 and TensorRT 7.2 on T4.
Here, we use Torch1.8.0 inference on V100 and TensorRT 7.2 Cuda 10.2 Cudnn 8.0.2 on T4.

## 2. Reproduce speed

Expand Down

0 comments on commit 2e1c276

Please sign in to comment.