add pynvml

This commit is contained in:
Somdev Sangwan 2023-06-04 13:24:32 +05:30 committed by GitHub
parent 289bcff9a3
commit 2606be04ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -13,4 +13,5 @@ tensorflow==2.13.0rc1; sys_platform == 'darwin'
tensorflow==2.12.0; sys_platform != 'darwin'
opennsfw2==0.10.2
protobuf==4.23.2
pynvml==11.5.0
tqdm==4.65.0