Pytroch入坑 1. Deep15.5 + CUDA8.0 + pytroch环境搭建

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/

在此网页选择自己适合的安装文件

我下的是torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl,最好选择带mu的,我开始选带m的不兼容

下好后执行

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,照做就行。安装完这个就能跑大多数代码了~


  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值