0.前言:
一个Pytroch简单入门系列,配置 i5-6200u+ gt940m+4G内存+deepin15.5.
Ubuntu应用虽广,但是配置环境好麻烦....以后用熟了再换吧....小白上手即用推荐Deepin,什么都弄好了。
推荐安装教程网址:https://blog.csdn.net/puzzleofmyheart/article/details/79183094(台式机)
https://blog.csdn.net/liaodong2010/article/details/71482304
1.安装nvidia驱动
现在绝大多数笔记本都是双显卡,本文推荐的方法适合绝大多数笔记本用户,台式机参照上面的网址。
1.安装nvidia-bumblebee
sudo apt update
sudo apt install nvidia-smi nvidia-driver
一行命令搞定nvidia驱动、bumblebee切换程序、和显卡状态监控程序。
不用管nouveau驱动,系统会自己屏蔽掉。
然后重启
sudo reboot
检测一下
需要注意,安装完nvidia-bumblebee后,以后针对显卡的操作前面都要加上 optirun
2. 安装cuda驱动
deepin方便之处来了。
如果式cuda8.0及其以下,deepin自带的gcc版本高了,需要进行降级。
sudo apt install g++-4.8 gcc-4.8
cd /usr/bin
sudo rm gcc g++
sudo ln -s g++-4.8 g++
sudo ln -s gcc-4.8 gcc
这个操作过程中可能会报错,无视,执行完就行
安装cuda的开发工具
sudo apt install nvidia-cuda-dev nvidia-cuda-toolkit nvidia-nsight nvidia-visual-profiler
sudo apt install python-pycuda
python-pycuda 是cuda的python开发接口。
3. 下载pytorch
到pytorch的官网上去下载pytorch 的包
一种是根据自己的需要,选择安装命令再终端中自动下载安装,但是国内环境较差,比如我这个连命令都刷不出来,所以选择离线安装的方式:
http://pytorch.org/previous-versions/
在此网页选择自己适合的安装文件
sudo pip install -0.3.0.post4-cp27-cp27mu-linux_x86_64.whl
pytroch 若安装失败,则可能是版本不对、pip没升级、numpy没升级,百度一下就行
4.测试
5.torchvision安装
这是官方的用来下载模型,加载数据的,很方便
我用 pip安装失败,最后参照这篇博客 https://blog.csdn.net/guotong1988/article/details/79065646
git clone https://github.com/pytorch/vision.git
git checkout 0.2.0
pip install pillow
python setup.py install
可能会让你升级pip,照做就行。安装完这个就能跑大多数代码了~