search

PRIS-CV / AP-CNN_Pytorch-master

Weakly Supervised Attention Pyramid Convolutional Neural Network for Fine-Grained Visual Classification (TIP2021)
MIT License
46 stars 4 forks source link

BugReport #8

Open RichardcLee opened 2 years ago

RichardcLee commented 2 years ago

1.You should declare the operating system and GPU environment. Torch0.4.1 + CUDA9.0 will cause an error if run your code on a gpu with a version higher than 2080Ti. Switch to CUDA9.2 to run through the code.

  1. lib pyyaml must >= 5.1
One-people-world commented 1 year ago

hello , which version of CUDA is select if my code on a gpu with 2080Ti ?
Thanks.

One-people-world commented 1 year ago

what's more, I can't find the download source for CUDA9.0, the download source is not available