ubuntu18.04 pytorch 深度学习环境搭建
一、工具安装 sudo apt-get install gcc sudo apt-get install g++ sudo apt-get install make二、禁用nouveau(必须的) sudo gedit /etc/modprobe.d/blacklist.conf 添加 blacklist nouveau sudo update-initra...
一、工具安装
sudo apt-get install gcc
sudo apt-get install g++
sudo apt-get install make
二、禁用nouveau(必须的)
sudo gedit /etc/modprobe.d/blacklist.conf
添加 blacklist nouveau
sudo update-initramfs -u
sudo reboot
可以通过 lsmod | grep nouveau 检查是否已禁用。
三、安装cuda(装nvdia显卡时,确定不在lightdm环境下,gdm3可以)
1.如果在lightdm下,停止lightdm串口管理器,之后会进入黑窗口,建议停止之前拍照接下来的操作。
sudo service lightdm stop (or:sudo systemctl stop lightdm)
2.清除旧版本的nvidia驱动和cuda:(推荐安装使用cuda***.run文件安装驱动和cuda;也可以单独安装驱动,在run文件安装过程去除驱动安装,但是得保证版本对应)
通过run文件安装:
sudo /usr/local/cuda-10.2/bin/cuda-uninstaller
sudo /usr/bin/nvidia-uninstall
通过apt安装:
sudo apt-get --purge remove "*cublas*" "cuda*"
sudo apt-get --purge remove "*nvidia*"
3.下载cuda(推荐run文件)
sudo sh cuda**.run
4.查看显卡信息,验证驱动是否安装成功
nvidia-smi
5.sudo service lightdm start
6.cd usr/local
查看具体的路径,上面两个图圈住的名字不一样,对应自己的版本,更改下面export的内容。
7.sudo gedit ~/.bashrc
添加行(如果不加nsight-compute路径,使用nsys会出现没有此命令)
export PATH=/usr/local/cuda-10.2/bin:/usr/local/cuda-10.2/nsight-compute-2019.5.0/${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64/${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
8.source ~/.bashrc
9.检查是否安装成功的几个方法:
1.nvcc -V
2.cat /proc/driver/nvidia/version
3.cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
4.nsys --version
四、安装cudnn
可以官网下载deb的文件,直接双击安装,但是需要注册账号
五、安装pytorch
进入官网下载即可。
注:如果网络不好,建议使用pip安装,手动在浏览器下载pytorch的whl文件(700M,网速不好很容易中断),然后本地安装。
手动安装whl文件的方法:
1.安装依赖库 conda install numpy matplotlib py-opencv
2.pip install /home/liu/package/pytorch/torch-1.1.0-cp37-cp37m-linux_x86_64.whl
六、检查是否安装成功
import torch
print(torch.__version__)
print(torch.cuda.is_available())
conda list
pip list
参考:
官方linux安装教程:
网页版:
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐
所有评论(0)