Ubuntu系统开发环境配置(二)

上一篇我们介绍了Ubuntu系统针对前端开发必备的环境配置,这篇文章我们介绍python的开发环境。
最近我也在研究机器学习和深度学习,配置环境的时候踩了不少的坑,这篇总结写给想要再机器学习方面有探索的同学。

python环境

可以参考我另一篇文章 快速搭建python环境

pycharm安装

pycharm是python的开发IDE,这里介绍在Ubuntu上安装的方法
pycharm官网


选择Community的下载,完全够用且免费
下载下来后是一个压缩包,使用解压工具解压,移动到/opt/pycharm 文件夹中
然后执行

$ sh /opt/pycharm/bin/pycharm.sh

可以随便打开一个文件夹
如何放置到收藏栏呢?



点击 tools -> create desktop entry
然后在软件库中就有了这个图标,右键放置收藏栏,就可以了。

GPU

机器学习离不开gpu,而在Ubuntu系统上配置一块显卡,有以下一些工具要安装。

显卡驱动安装

$ ubuntu-drivers devices

这里选择 nvidia-driver-515 执行安装命令

$ sudo apt install nvidia-driver-515

重启后验证是否安装成功

$ nvidia-smi

显示驱动已经安装完成

cuda安装

cuda是可以调用显卡进行计算的接口工具包,一定要选择和所选驱动一致的大版本,不能高于驱动版本。以下是最新的对照表


从这里选择你需要的cuda版本
这里我选的cuda11.7 这里下载


选择你的系统版本,installer Type 选择 runfile
执行下面提示的命令

$ wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
$ sudo sh cuda_11.7.0_515.43.04_linux.run

这个文件比较大,耐心等待...
执行安装后会出现以下界面


输入 accept

这里将Driver选项去掉,因为刚才已经安装了driver
然后install
修改 .bashrc 文件
export LD_LIBRARY_PATH=LD_LIBRARY_PATH:/usr/local/cuda-11.7/lib64 export PATH=PATH:/usr/local/cuda-11.7/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.7
验证是否安装成功

$ nvcc -V

显示上图中信息,说明cuda已经安装好了

pytorch环境安装

pytorch是当前比较流行的机器学习框架,这里介绍安装pytorch的方法
pytorch官网
选择和你系统相匹配的选项,如下图


因为我们上文已经安装过conda,所以这里可以选择conda安装

$ conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

耐心等待~~~

安装opencv

opencv也是我们在机器学习应用在计算机视觉时不可缺少的工具,这里采用pip安装

$ pip install opencv-python

尾声

以上是我在做机器学习环境准备积累的经验,希望能给后来者一些帮助。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容