> 文章列表 > 安装更新cuda因为torch安装失败

安装更新cuda因为torch安装失败

安装更新cuda因为torch安装失败

1、torch要求1.12+以上版本

>>> import torch
>>> torch.cuda.is_available()
//envs/deepspeed/lib/python3.9/site-packages/torch/cuda/__init__.py:88: UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 10010). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver. (Triggered internally at /opt/conda/conda-bld/pytorch_1666643016022/work/c10/cuda/CUDAFunctions.cpp:109.)return torch._C._cuda_getDeviceCount() > 0
False

2、再次卸载cuda及驱动。

链接作用不大,直接来这个。解决。

sudo /usr/bin/nvidia-uninstall<