question about the version of pytorch and cuda？

TmacMai / ARGF_multimodal_fusion

codes for: Modality to Modality Translation: An Adversarial Representation Learning and Graph Fusion Network for Multimodal Fusion

45 stars 11 forks source link

question about the version of pytorch and cuda？ #1

Closed wangjiwu closed 4 years ago

TmacMai commented 4 years ago

Our Pytorch version is 1.4.0 and CUDA version is 10.1

We also test the code in CUDA 9.0 and Pytorch 1.0.0, and it still works.

wangjiwu commented 4 years ago

thanks！ I try it in Pytorch 1.5.1 and CUDA 10.1. An error occurred when I was running the code in this environment. the code run well in the lower version 👍