search

microsoft / Oscar

Oscar and VinVL
MIT License
1.04k stars 251 forks source link

Run Oscar with Pytorch1.7 and Cuda10.1 #142

Closed rachs closed 2 years ago

rachs commented 2 years ago

Can I run Oscar with Pytorch 1.7 and cuda 10.1? I want to run the scenegraph benchmark repo to create object features and oscar to generate image captions from the same virtualenv. Has anyone done this?

jontooy commented 2 years ago

Hi rachs,

I'm working on a similar project and have spent alot of time on installing environments for Oscar. Their INSTALL.md file indeed states requirements

I've managed to install both Oscar and sgg_benchmark using the same version of Pytorch (1.7.1) and Cuda (10.1). See this demo repo, so I think it might be possible to run them on the same virtualenv! Some files, and pretrained models might have to be downloaded from the respective repos in order to make the demos work.

Hope it helped! Good luck with your project and let me know how it goes :)

rachs commented 2 years ago

Didn't see this till now, thank you jontooy! Yes it worked for me with cuda 10.1 too.