从RTX2080Ti 换成 RTX3090 的驱动、cuda11.1版本和pytorch安装问题

发布网友 发布时间:17小时前

我来回答

1个回答

热心网友 时间:8分钟前

感谢“暮晓”在评论区的提醒,RTX3090显卡在CUDA11.0环境下并未完全支持,真正全面支持的是CUDA11.1版本。因此,在从RTX2080Ti升级至RTX3090后,更新驱动和CUDA版本成为了必要的操作。

在Ubuntu16.04系统中,安装RTX3090时需确保CUDA11.1及以上版本与PyTorch1.7.1或以上版本兼容,以充分利用GPU资源。昨晚至今,我正为在CUDA11.0环境下重新安装PyTorch1.7所遭遇的问题而苦恼。

首先,获取最新的NVIDIA驱动版本460.32.03(或更新版本),请访问官网下载链接Download/driv...>。

其次,CUDA版本建议选择11.0或更高版本,推荐使用11.1或以上版本,以确保与RTX3090显卡的兼容性。

对于CUDA的配置,建议在conda环境中进行,这样能自动完成所需的配置操作。

然而,PyTorch的安装过程中遇到下载速度问题,原镜像下载速度缓慢,因此切换至国内清华源。

在`./condarc`文件中添加清华源路径后,PyTorch文件夹资源未被正确提取,问题未解决。

安装的PyTorch版本总是为1.3的CPU版本,无法实现CUDA调用。

解决方法是直接定位到清华源链接。

在conda虚拟环境的命令窗口,输入以下命令

声明声明:本网页内容为用户发布,旨在传播知识,不代表本网认同其观点,若有侵权等问题请及时与本网联系,我们将在第一时间删除处理。E-MAIL:11247931@qq.com