安装更新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
<