搭建Linux服务器环境

转载自:https://www.zhihu.com/people/an-xing-le-20/posts


算法开发环境搭建

一. CUDA 安装

1. 驱动安装

首先需要自己的GPU型号,这里执行

  lspci | grep -i vga

然后我们可以进入NVIDIA官网查看对应的驱动,并下载。

img

2. 下载CUDA

我们进入cuda toolkit 下载最新cuda开发包(请与 pytorch、tensorflow 版本相对应)。

img

3. 安装

下载的时候,实际已经可以看到安装方式了:

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.runsudo sh cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run

但是可能还是会遇到一些坑,多是一些权限等问题,推荐超级管理员安装。

4. cuDNN安装

进入cuDNN介绍目录http://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html。找到和CUDA对应的cudnn开发包。

// 切换到压缩包所在位置,解压
$ tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz

// 复制 文件到cuda 目录下
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

// 修改文件权限
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

二. Python环境安装

1. 下载

首先需下载 Anaconda开发包,请登录 Anaconda官网 或者执行

wget https://repo.anaconda.com/archive/Anaconda3-2019.07-Linux-x86_64.sh

来下载安装二进制文件。

2. 安装

本着linux用户“谁使用,谁负责”的原则,用户在本目录下直接执行

sh Anaconda3-XXX.sh

然后一路默认 “yes” (可以不安装Vscode)即可(注意:这里需要仔细查看,眼镜不要离开屏幕)。

3. 验证使用

bash下输入python,可看到python运行环境应该为:

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-0LyIMcAC-1588772568523)(http://10.30.1.6:33511/uploads/file_image/ueditor/php/upload/image/20190924/1569308459900916.png)]

具体版本号等视自己安装情况而定。

4. 安装虚拟环境

不同的项目往往需要安装多个不同的python解释器,不同的python开发包。 为方便进行多python环境管理、开发(各个项目的python环境互不干扰),开发环境需要多套不同的python解释器以及开发包与项目一一对应。这个时候python虚拟环境安装管理就极为便捷。

本着简洁易用原则,我们使用conda来进行多python虚拟环境的管理。

// 安装 版本为X.X的python解释器
conda create -n your_environment_name(具体是你的python虚拟环境名) python=X.X(这里替换为你的python版本,例: 2.7, 3.6, 3.7...)
// 激活虚拟环境(开始使用该环境, 项目开发,包管理....)
conda activate your_environment_name

5.修改文件

打开文件.bash_profile,添加如下

# cuda10.0
export PATH=$PATH:/usr/local/cuda-10.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export LC_CTYPE ='zh_CN.utf8'

三. pytorch、tensorflow 环境安装

这个如果不会,请直接回大学读大一吧。地球已经拯救不了你了。

这里注意两点:更新conda版本和下载源添加清华
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
安装得时候不要加 -c参数
conda install pytorch torchvision cudatoolkit=10.0

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值