700字范文,内容丰富有趣,生活中的好帮手!
700字范文 > anaconda重新安装pytorch 使用GPU加速

anaconda重新安装pytorch 使用GPU加速

时间:2024-06-24 14:52:33

相关推荐

anaconda重新安装pytorch 使用GPU加速

前言

之前由于要使用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加速了

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。