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