1. 第一步首先安装cuda,这里安装cuda10.1版本。
cuda官网:CUDA Toolkit Archive | NVIDIA Developer
下载好上传到服务器中
sudo sh cuda_10.1.243_418.87.00_linux.run
按空格取消Driver安装,往下选择Install安装
配置环境变量:
sudo vim ~/.bashrc
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}