下载cuda和cudnn
1 | wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run |
下载cudnn,下载需要官网上点击,需要登录,并且获得的链接无法使用wget
下载
解决:使用https://developer.download.nvidia.com/compute/redist/cudnn/
代替,以cudnn-11.2为例:
1 | wget https://developer.download.nvidia.com/compute/redist/cudnn/v8.1.0/cudnn-11.2-linux-x64-v8.1.0.77.tgz |
配置CUDA
开始安装
给cuda授权
1
chmod +x cuda_xxxx_linux.run
运行run文件
1
sh cuda_xxxx_linux.run
选择
Continue
进入输入
accept
后回车同意使用
上下键
和回车勾选对话框,只安装CUDA Toolkit选择
Options
并进入,配置Toolkit Options
和Library install path
取消勾选其它项,修改Toolkit安装路径
输入对应的路径如:
回车,选择
Done
回到Options
菜单,选择Library install path
输入对应路径如:
回车,选择
Done
回到主菜单,选择Install
环境变量配置
输入
vim ~/.bashrc
添加:1
2
3export CUDA_HOME=$CUDA_HOME:/home/xxx/cuda-11.2
export PATH=/home/xxx/cuda-11.2/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/xxx/cuda-11.2/targets/x86_64-linux/lib输入
source ~/.bashrc
启用环境变量
配置cuDNN
解压cuDNN
1
tar -zxvf cudnn-11.2-linux-x64-v8.1.0.77.tgz
将解压的文件移动到CUDA安装目录
1
2mv cuda/include/cudnn.h cuda-11.2/include
mv cuda/lib64/libcudnn* cuda-11.2/targets/x86_64-linux/lib/修改权限
1
chmod a+r cuda-11.2/include/cudnn.h cuda-11.2/targets/x86_64-linux/lib/libcudnn*
PyCharm远程调用的问题
如果是PyCharm远程调用,不会执行.bashrc
中的环境配置,因此还需要自己配置环境
点击Edit Configration
选择对应的运行环境配置环境变量
最后发现了新的更简单的方法
去conda找对应版本的环境包
使用conda安装,例如:
1
conda install -c jjhelmus tensorflow-gpu-base
完成