说明:本配置以Ubuntu系统为例进行说明。
1. 安装NVIDIA驱动,安装完毕可以在终端使用“nvidia-smi”命令进行检测和查看版本。
2. 安装CUDA Toolkit,根据操作1中的命令可以看到cuda driver的版本,根据该版本,搜索“cudatoolkit”+“版本号”,基本上百度搜索出来的第一个就是,或者直接在NVIDIA官网查找。
3. 根据系统版本选择相应的CUDA Toolkit文件,并根据安装说明里的指令,进行安装,安装完毕可以在终端中使用“nvcc -V”进行检测及查看安装的版本信息。
4. 激活环境变量,在终端里使用“gedit ~./bashrc”命令,进入bashrc文件,在文件最后位置写入(针对自己的版本进行路径修改)
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64 export PATH=$PATH:/usr/local/cuda-10.2/bin export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.2
修改完成后保存。
5. 在终端中使用命令“source ~./bashrc”激活环境变量。