前提:已安装显卡驱动、cuda和cudnn。
传统安装方法由于防火墙的原因,需要翻墙才能安装。这里采用镜像安装的方式:
- 添加镜像
#首先先添加清华的镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
vi ~/.condarc (vi 常用命令:i –编辑模式;esc – 退出编辑进入只读模式;shift+ZZ 保存并退出(在只读模式下))
channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
show_channel_urls: true
- 修改环境变量
vi ~/.bashrc
添加变量如下图所示
source ~/.bashrc #更新环境变量
nvcc -V #查看cuda版本信息
# .bashrc
# User specific aliases and functions
alias rm='rm