ubuntu 机器学习环境搭建

ubuntu 机器学习环境搭建

第一步下载官方驱动

官方驱动 | NVIDIA

你也可以选择使用命令行安装驱动

sudo apt update
sudo apt install 

第二步下载 cuda cudnn

阿里云网盘下载

NVIDIA阿里云盘下载

官方下载

CUDA Toolkit Archive | NVIDIA Developer

cudnn是需要注册才可以下载的

cudnn

安装cuda和cudnn

chmod +x ./cuda*.run 
sudo ./cuda*.run --override
sudo apt install ./cudnn*.deb

配置环境变量

pytorch安装

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116