安装NVDIA驱动的前提
sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
# 关闭图形界面
systemctl isolate multi-user.target
# xshell方式登录输入下面命令(tty模式)输入密码不正确最好使用字母上面的数字
modprobe -r nvidia-drm
# 开启图形界面
systemctl start graphical.target
安装CUDA Toolkit的网址
按照CUDA Toolkit安装过程中遇到的问题
错误描述:“media change: please insert the disc labeled” when trying to install Ruby on Rails [duplicate]
解决网址
CUDA安装成功之后的测试程序
头文件部分foo.cuh
#ifndef FOO_CUH
#define FOO_CUH
#include <stdio.h>
__global__ void foo();
extern "C"
void useCUDA();
#endif
源程序foo.cu部分
#include "foo.cuh"
#define CHECK(res) { if(res != cudaSuccess){printf("Error :%s:%d , ", __FILE__,__LINE__); \
printf("code : %d , reason : %s \n", res,cudaGetErrorString(res));exit(-1);}}
__global__ void foo()
{
printf("CUDA!\n");
}
void useCUDA()
{
foo<<<1,5>>>();
CHECK(cudaDeviceSynchronize());
}
main.cu源文件部分
#include <stdio.h>
#include <iostream>
//#include <cuda_runtime.h>
#include "foo.cuh"
int main()
{
std::cout<<"Hello NVCC"<<std::endl;
useCUDA();
return 0;
}
输入命令
nvcc -o main.out main.cu foo.cu