I just made it work now! I checked the developed post (https://github.com/pytorch/pytorch/issues/494), and some guy made it work and I installed it from his source. Only cuda80 works in my desktop. And also thanks for your reply!
1 Like