前言
之前由于要使用pytorch,所以就一股脑儿地安装了。安装方法如下(当时安装pytorch的时候我没有管那个CUDA版本,以为不碍事):
/qq_43391414/article/details/109672442
然后,用了一会儿torch之后,才发现要开始用gpu,但是python检查我的电脑后,告诉我:不能用。
报错:
The NVIDIA driver on your system is too old (found version 9010).Please update your GPU driver by downloading and installing a newversion from the URL: /Download/index.aspxAlternatively, go to: to installa PyTorch version that has been compiled with your versionof the CUDA driver.
然后我就去下载最新的驱动。
/download/index.aspx?lang=en-us#
在上面的表格填入你的显卡参数,这个你需要自己查一下自己电脑的。
很简单,打开设备管理器-》显示适配器就可以看到
在官网中的那个表单填写晚了之后,点击search,下载驱动,并安装就可以了。
但是,在以为万事大吉了之后,却发现还是不行
这个时候才想到可能是cuda的版本不匹配。因为当初第一个链接安装pytorch的时候,那个是基于cuda10.2版本,而我的可能不是。下面去查看。
在桌面右键-》NVIDIA控制面板,打开如下,点击系统信息。
发现是10.1版本的。
所以要去更新到10.2版本。还是上面更新驱动的那个网址。点击如下:
由于流量告急,我没选择将cuda版本从10.1提升到10.2,而是重新装了pytorch。因为重新安装只需要1G,而且感觉更省事。
重装的方法:
先删除原版,打开anaconda的命令行终端,输入:
conda uninstall cudatoolkit
我是这么做的,但是其实我猜输入:
conda uninstall pytorch
估计上述是一样的,因为当初安装的时候,这些是绑定在一起安装的,你现在要删除的时候,anaconda会自动检测到这些关联在一起的,所以会一起删除。
输入y就可以将这些一起全部删除了,然后重新安装cudatoolkit=10.1的版本。至于如何安装请参考博文:anaconda安装pytorch
最后,激动人心的时刻:
之前由于torch和我的电脑cuda版本不一样,一直是False。
可以开始使用GPU加速了