RenYurui / Global-Flow-Local-Attention

The source code for paper "Deep Image Spatial Transformation for Person Image Generation"
https://renyurui.github.io/GFLA-web
Other
568 stars 84 forks source link

nvcc fatal : Unsupported gpu architecture 'compute_60' #79

Closed Mathilda88 closed 3 years ago

Mathilda88 commented 3 years ago

I'm running linux: Python:3.6.4 Pytorch 1.1.0 Torchvision: 0.2.0 Cuda: 9.0 I also tried 8 and 10 GPU: Tesla V100 Could anyone share the environment setting for a correct adjustment? I keep running into nvcc fatal : Unsupported gpu architecture 'compute_60', totally solicited!!!!!