发布网友 发布时间: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虚拟环境的命令窗口,输入以下命令