Win10下cuda9.0安装正确,torch.cuda.is_available()为False的解决方法

先说一下,我遇到的问题比较复杂,因为这个东西本身很简单,但是遇到一些突发情况,也是头疼,好在百度百度百度。。。

 

 

0、前提是,我电脑本身就已经有tensorflow的gpu版本并安装cudn9.0的版本,且我用了挺久了,这次我要安装pytorch的gpu版本。

 

1、大概思路就是按照tensorflow之前cudn9.0和cudnn一样的pytroch的gpu版本就可以了,甩个链接https://blog.csdn.net/Mind_programmonkey/article/details/99688839,但是这个链接对第一次下载cudn和cudnn的有用,对我就没用了,因为我都有了cudn和cudnn,直接用就好了。也就是tensorlfow和pytorch共用一个cudn9.0。

 

2、但是安装之前需要在anaconda下重新创建一个虚拟环境,虚拟环境下安装相应版本,接着甩一个链接https://pytorch.org/

 

3、重点来了,我安装后发现重启几次仍然失败,就是这样:输入:torch.cuda.is_available() 得到:False

 

4、之后我将显卡驱动更新到最新版后,重启输入torch.cuda.is_available()  得到:True

甩2个链接:

https://blog.csdn.net/qq_33312515/article/details/90175299

https://jingyan.baidu.com/article/f0e83a2556f03e22e5910123.html

 

 

 

 

 

你可能感兴趣的:(神经网络)