M40 安装cuda和torch
- 驱动
- sudo add-apt-repository ppa:graphics-drivers/ppa
- sudo apt update
-
sudo apt install nvidia-driver-550
- CUDA 11.8
- https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=runfile_local
- Pytorch
- https://pytorch.org/get-started/previous-versions/
- pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu118