Я работаю над проектом, использующим PyTorch с CUDA, и в моей среде разработки все работает гладко. Однако когда я упаковываю проект в файл .exe с помощью PyInstaller, размер файла составляет 4,5 ГБ, причем только на PyTorch приходится около 4 ГБ этого размера. Кто-нибудь знает эффективные способы управления зависимостями PyTorch при создании файла .exe для уменьшения конечного размера? Будем очень признательны за любые советы!
Код: Выделить всё
from torchvision.models import mobilenet_v2,MobileNet_V2_Weights
import torchvision.transforms as transforms
from torch.nn import Sequential
from torch.cuda import is_available
from torch import no_grad,load
это импорт, для которого потребовалось только 4 ГБ памяти.
Подробнее здесь:
https://stackoverflow.com/questions/791 ... -with-cuda
Мобильная версия