0. 环境要求
安装好Nvidia驱动的Ubuntu服务器或者WSL2(后者一般自带显卡驱动)
- 安装好Nvidia驱动的Ubuntu服务器或者WSL2(后者一般自带显卡驱动),本文以WSL2为例
通过nvidia-smi
验证是否安装好驱动,以及所安装的驱动和cuda的版本
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-pQaywjO4-1660648016311)(image_1.daccfac4.png)]
如果没有安装好驱动可以尝试以下命令或自行百度安装
sudo apt update
sudo ubuntu-drivers devices # 查看显卡型号和推荐的驱动 sudo ubuntu-drivers autoinstall # 自动安装推荐的版本驱动
1. 安装Docker
- 卸载旧版
sudo apt-get remove docker docker-engine docker.io containerd runc
- 安装依赖
sudo apt-get update
sudo apt-get install ca-certificates curl gnupg lsb-release