跑模型的时候遇到这个问题,“cuda is not available, please set 'gpu' -1”,查了网上没有相关的解决办法,本电脑的NVIDIA是10.1版本,请问有解决的办法吗
你安装的pytorch是什么版本的呀,需要跟你的NVIDIA对应才可以
你的cudnn呢?这个没安装也是不行的,pytorch版本需要和cuda版本对应,具体对应关系可以在官网找到