##配置Ubuntu相关深度学习环境 在配置深度学习的时候,我们往往只配置一次,但是一旦出了问题,重新配置的时候总会出现很多问题。在此,总结了在服务器中配置相关环境中一些环节。希望能帮助到大家!
- 准备ubuntu系统
- 安装合适显卡驱动
- 安装cuda和cudnn
- 安装anaconda
- 配置深度学习env
1、安装系统 2、安装合适的显卡驱动: 在我们跑代码的时候,出现torch.cuda.is_availble() == false。该问题出现的一种原因是驱动对应cuda版本低于我们安装的cuda版本,这时候我们需要更新驱动。当然刚开始安装驱动的步骤是一样的。 备注:最新版本的驱动最好,适用于当前所有版本的cuda。 2.1. 查看自己的显卡名称
~$ lspci | grep -i nvidia
2.2. 在nvidia官网下载驱动程序nvidia官网,
2.3. 卸载旧版本驱动,如果是新安装驱动,可以取消该操作 sudo apt-get purge nvidia*
禁止系统自带的nouveau驱动
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
blacklist nouveau
options nouveau modeset=0
sudo update-initramfs -u
lsmod | grep nouveau
2.3 开始安装 在终端上,进入下载的驱动程序.run目录下。进行如下操作
sudo chmod +x NVIDIA-Linux-x86_64-440.36.run
sudo ./NVIDIA-Linux-x86_64-440.36.run
开始安装后,都是默认设置,一路按 enter ,直接结束。 查看显卡驱动是否安装成功
nvidia-smi
3、安装cuda和cudnn cuda和cudnn是深度学习中必须的模块,安装成功与否关系到是否可以用GPU。 安装cuda和cudnn教程
4、安装anaconda 转载自: Ubuntu安装anaconda 其中重要注意的是需要配置相应的环境变量
5、配置个人的环境
conda create -n envname python=[version]
source activate [envname]
conda config --show
conda list
conda env list
conda config --add channels https:///
conda config --remove channels https://
vim ~/.condarc
当前配置的镜像源无法下载相关的包时,用临时的镜像源下载相关package。
pip install opencv-python torchvision -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
pip install opencv-python torchvision -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com#临时换其他镜像源网站
常用的国内镜像源: 连接1 连接2
待续:安装过程中的常见错误
|