本文记录安装过程,多为自用。
一、CUDA的安装 1、查看操作系统先安装
sudo yum install -y redhat-lsb
输入命令查看自己的操作系统:
lsb_release -a2、查看显卡
使用命令查看显卡:
lspci | grep -i nvidia
可以看到如下信息:
3、下载CUDA去官网下载需要的CUDA:
https://developer.nvidia.com/cuda-toolkit-archive
本文下载的是11.1版本:
输入官方命令安装。
设置环境变量:
在主目录下有.bashrc文件,打开.bashrc:
vim ~/.bashrc
在文件末尾添加以下命令:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64 export PATH=$PATH:/usr/local/cuda/bin export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
保存退出之后执行:
source ~/.bashrc4、验证
使用命令验证安装:
nvcc -V二、cudnn的安装
去官网下载对应的cudnn版本,我这里下载的是cudnn v8.0.4,下载之后上传:
之后使用命令解压:
tar -xzvf cudnn-11.1-linux-x64-v8.0.4.30.tgz
会得到新目录cuda,进入cuda目录下:
需要关注的是include和lib64这两个目录,分别把他们复制到cuda驱动目录:
sudo cp cuda/include/* /usr/local/cuda/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ sudo chmod a+r /usr/local/cuda/include/cudnn.h sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
验证一下:
nvidia-smi
完成!
参考:
https://zhuanlan.zhihu.com/p/339062791
三、Anconda的安装 1、安装参考博文:
https://blog.csdn.net/Lin1169404361/article/details/123288482
2、卸载主要是因为安装的目录内存不够了...
cd到anconda3目录下:
执行命令:
rm -rf anaconda3
有设置环境变量进入.bashrc文件删除有关anconda3设置的语句即可。
四、安装GIT 1、查看linux系统uname -a
本文是64位:
2、启用EPEL存储库wget http://dl.fedoraproject.org/pub/epel/epel-release-latest-7.noarch.rpm rpm -ivh epel-release-latest-7.noarch.rpm3、安装GIT
yum install -y git
完成!
参考:
https://blog.csdn.net/weixin_46269257/article/details/116245531
五、创建虚拟环境并激活 1、创建conda create -n torch python=3.72、激活
conda activate torch
完成!



