安装CUDA时需要和Torch版本对应起来,最好先去torch官网上确定要安装的torch版本对应的CUDA版本。
在安装CUDA之前需要先确定是否已经安装驱动,打开终端输入 nvidia-smi
,若有输出,则表明驱动安装过,否则需要先安装驱动(驱动安装教程)
一、CUDA11.1安装
1、CUDA11.1下载
先去CUDA官网上下载要安装的版本(CUDA11.1下载链接),依次选择Linux——》x86_64——》Ubuntu——》20.04——》runfile(local),根据自己的电脑配置选择即可
打开终端