M40 安装cuda和torch

设置BIOS

打开 PCIE 4G 支持 驱动

1. sudo add-apt-repository ppa:graphics-drivers/ppa
2. sudo apt update

3. sudo apt install nvidia-driver-550

CUDA 11.8

1. 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
2. sudo sh cuda*

Pytorch

1. https://pytorch.org/get-started/previous-versions/
2. pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu118