RTX显卡+pytorch报错:cublas runtime error: the GPU program failed to execute

参考了这篇文章:

https://blog.csdn.net/u013128836/article/details/101636736

主要是pytorch、cuda、之间的版本问题,我直接从pip源上安装的torch1.1和torchvision,会报上面的错,但是直接从官网下载的.whl文件重装pytorch立即解决

pytorch官网

你可能感兴趣的:(RTX显卡+pytorch报错:cublas runtime error: the GPU program failed to execute)