Requirements of pytorch #29

Activity

1. Chen-Xieyuanli commented on Mar 17, 2022

   

   Chen-Xieyuanli

   on Mar 17, 2022

   Member

   More actions

   @BIT-MJY could you please give your PyTorch setup here?
2. BIT-MJY commented on Mar 18, 2022

   

   BIT-MJY

   on Mar 18, 2022

   Last edited by BIT-MJY

   Collaborator

   More actions

   Hello @smalltheater ,

   Thanks for using our code! The setup of OverlapNet (pytorch) is the same as the one used in our newest work OverlapTransformer for faster retrieval, which is:

   torch==1.10.2+cu113  
   torchvision==0.11.3+cu113  
   torchaudio==0.10.2+cu113  
     
   matplotlib==3.4.3    
   numpy==1.20.3    
   tensorboardx==2.4    
   pyyaml==6.0    
   opencv-python==4.5.4.58    
   faiss-cpu==1.7.1    
   scikit-learn==0.24.2    
   

   You can download our Torch dependences by pip:

   pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio==0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

   or by conda:
   conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

   By the way, the data used for training and evaluation here is the same as the data used in OverlapTransformer, maybe our new work can also help you.
3. Chen-Xieyuanli

   added

   good first issueGood for newcomersGood for newcomers

   on Mar 21, 2022
4. Chen-Xieyuanli commented on Mar 21, 2022

   

   Chen-Xieyuanli

   on Mar 21, 2022

   Member

   More actions

   @BIT-MJY Thanks for the detailed reply.