RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with
更新 NVIDIA 驱动可以通过以下步骤完成:
-
检查可用的 NVIDIA 驱动版本: 首先,您可以查看当前可用的 NVIDIA 驱动版本。打开终端并运行以下命令:
bashCopy code
ubuntu-drivers devices
这将显示您的系统上可用的 NVIDIA 驱动版本列表。
-
选择要安装的驱动版本: 根据上一步的输出,选择要安装的 NVIDIA 驱动版本。通常,建议选择最新的稳定版本。
-
安装所选的 NVIDIA 驱动: 使用以下命令来安装所选的 NVIDIA 驱动。将
<version>
替换为您选择的驱动版本。bashCopy code
sudo apt install nvidia-driver-<version>
例如,如果您要安装版本 460,您可以运行:
bashCopy code
sudo apt install nvidia-driver-460
-
重启系统: 驱动程序安装完成后,最好重启系统以使更改生效。
-
验证驱动安装: 安装完成后,您可以运行以下命令来验证 NVIDIA 驱动是否已正确加载:
bashCopy code
nvidia-smi
如果该命令显示有关您的 NVIDIA GPU 的信息,则表示驱动已成功安装。
请注意,如果您之前已经安装了其他版本的 NVIDIA 驱动,您可能需要在安装新驱动之前卸载现有的驱动。可以通过运行以下命令来卸载驱动:
bashCopy code
sudo apt remove --purge nvidia-*
然后按照上述步骤安装新的驱动版本。
ubuntu-drivers devices Command 'ubuntu-drivers' not found, but can be installed with: sudo apt install ubuntu-drivers-common
系统中缺少 ubuntu-drivers
命令。您可以按照终端中的建议安装 ubuntu-drivers-common
来获取此命令。请按照以下步骤执行:
-
运行以下命令以安装
ubuntu-drivers-common
:bashCopy code
sudo apt install ubuntu-drivers-common
-
安装完成后,再次运行以下命令以查看可用的 NVIDIA 驱动版本:
bashCopy code
ubuntu-drivers devices
这样应该可以显示出您系统中可用的 NVIDIA 驱动版本。然后,您可以根据之前的步骤选择并安装适当的驱动版本。
ubuntu-drivers devices ERROR:root:aplay command not found == /sys/devices/pci0000:00/0000:00:1e.0 == modalias : pci:v000010DEd00001EB8sv000010DEsd000012A2bc03sc02i00 vendor : NVIDIA Corporation model : TU104GL [Tesla T4] driver : nvidia-driver-525-server - distro non-free driver : nvidia-driver-470-server - distro non-free driver : nvidia-driver-418-server - distro non-free driver : nvidia-driver-535-server - distro non-free driver : nvidia-driver-470 - third-party non-free recommended driver : nvidia-driver-450-server - distro non-free driver : xserver-xorg-video-nouveau - distro free builtin
获取了可用的 NVIDIA 驱动版本列表。根据您提供的输出,以下是一些可供选择的 NVIDIA 驱动选项:
nvidia-driver-525-server
nvidia-driver-470-server
nvidia-driver-418-server
nvidia-driver-535-server
nvidia-driver-470
nvidia-driver-450-server
xserver-xorg-video-nouveau
您可以根据您的需求和系统兼容性选择其中一个版本进行安装。例如,如果您想要最新的服务器级别驱动,您可以选择 nvidia-driver-535-server
。您可以使用以下命令安装选定的驱动版本(以 nvidia-driver-535-server
为例):
bashCopy code
sudo apt install nvidia-driver-535-server
安装完成后,您可能需要重新启动系统以使新的驱动生效。