centos下通过conda安装pytorch
一、安装anaconda
anaconda安装简单,只要确定自己的系统即可,具体安装请参考这里
二、确定自己的系统版本
我的是centos
cat /etc/redhat-release
查看linux系统方法:
# 方法一 cat /proc/version# 方法二 lsb_release -a# 方法三 cat /etc/issue# 方法四 cat /etc/redhat-release
三、确定自己的CUDA版本
什么是cuda:
CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题
简单说强大的芯片如果只是作为显卡就太浪费了,因此N卡厂商推出CUDA,让显卡可以用于图像计算以外的目的
什么是cudnn:(注意:CUDA和CuDNN一般都是结伴而行,如果需要CUDA,必然会需要CuDNN)
下载地址:注意这里需要注册才能下载,我用qq邮箱,密码首字母大写
CUDA看作是一个工作台,上面配有很多工具,如锤子、螺丝刀等。cuDNN是基于CUDA的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具,比如它就是个扳手
但是CUDA这个工作台买来的时候,并没有送扳手。想要在CUDA上运行深度神经网络,就要安装cuDNN,就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作,工作速度相较CPU快很多
cudnn下载解压后cuda目录下有下面内容:
修改权限:
sudo cp include/cudnn.h /usr/local/cuda/include sudo cp lib64/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
Centos7安装NVIDIA显卡驱动和CUDA10.1(以10.1为例):
先查看自己的linux上显卡型号:
lspci | grep -i nvidia
可见我的显卡杠杠滴
显卡驱动下载:(下载与自己显卡型号一致的驱动)
注意:在安装显卡驱动的时候如果报错
ERROR: An NVIDIA kernel module 'nvidia-uvm' appears to already be loaded in your kernel. This may be because it is in use (for example, by the X server), but may also happen if your kernel was configured
我们需要查看是否有程序占用(如果存在占用,请停掉该程序)
lsof | grep nvidia.uvm
当然显卡驱动也可以这样安装:(推荐)
sudo yum install nvidia-detect # 安装检查显卡的程序 nvidia-detect -v # 查看需要的显卡驱动型号 yum -y install kmod-nvidia # 安装驱动
卸载驱动:
sudo yum remove kmod-nvidia
如果通过官网下载驱动安装时找不到kernel,通过下面的方式安装驱动(确保kernel已经安装)
sudo ./NVIDIA-Linux-x86_64-430.40.run --kernel-source-path=/usr/src/kernels/3.10.0-514.el7.x86_64
cuda下载:
检查linux是否安装了GPU
lspci | grep -i nvidia
安装 cuda 的时候,会询问是否安装显卡驱动,说明 cuda 安装程序里包含了的显卡驱动;建议先不要安装 cuda 里的显卡驱动,待安装完 cuda 后,执行例子程序,如果报错再检查显卡驱动是否正确,避免覆盖原来的显卡驱动
安装gcc、g++编译器:先查看是否已经安装(gcc -v)
yum install gcc yum install g++
安装kernel-devel和kernel-headers:
yum install kernel-devel yum install kernel-headers
注意这里的kernel要和自己的linux系统版本一致:可以使用uname -r查看自己的系统版本
uname -r #3.10.0-514.el7.x86_64
有时候直接yum安装的版本比较新,导致显卡驱动安装失败
sudo yum remove kernel-devel # 卸载新的 sudo rpm -i kernel-devel-3.10.0-514.el7.x86_64.rpm #下载安装适合的版本
版本下载
编辑grub文件
vim /etc/default/grub
在GRUB_CMDLINE_LINUX
中的末尾(注意要包含在引号内)添加/rd.driver.blacklist=nouveau nouveau.modeset=0
随后生成配置:
grub2-mkconfig -o /boot/grub2/grub.cfg
禁用nouveau:将blacklist nvidiafb注释掉,并添加如下内容,禁用的原因点击这里
blacklist nouveau
options nouveau modeset=0
更新配置并重启:
sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak sudo dracut /boot/initramfs-$(uname -r).img $(uname -r) reboot
重启后确认禁用了nouveau:
lsmod | grep nouveau # 没有内容输出表示禁用成功
赋予权限并安装驱动和CUDA-10.0:
./NNVIDIA-Linux-x86_64-430.50.run ./cuda_10.1.243_418.87.00_linux.run
查看安装结果
nvidia-smi
显卡驱动和cuda安装成功后,就可以用conda install 来安装python包了(注意:虚拟机无法调用GPU,即使所有的环境配置好,代码依然会报:all cuda 忙或者不可用的错误)
推荐搭建使用docker来安装深度学习环境,方便部署
总结:大家可以参考这个来安装深度学习框架
转载于:https://www.cnblogs.com/always-fight/p/11512608.html
centos下通过conda安装pytorch相关推荐
- linux安装pytorch教程,centos下通过conda安装pytorch
一.安装anaconda anaconda安装简单,只要确定自己的系统即可, 二.确定自己的系统版本 我的是centos cat /etc/redhat-release 查看linux系统方法: # ...
- 使用conda卸载pytorch_centos下通过conda安装pytorch
一.安装anaconda anaconda安装简单,只要确定自己的系统即可, 二.确定自己的系统版本 我的是centos cat /etc/redhat-release 查看linux系统方法: # ...
- linux pytorch环境配置,linux下使用conda安装pytorch,并配置pytorch
8种机械键盘轴体对比 本人程序员,要买一个写代码的键盘,请问红轴和茶轴怎么选? 首先安装anaconda 去官网下载.sh包,安装之后会把jupyter一起安装好. 配置jupyter 首先先生成一个 ...
- conda安装pytorch下载太慢
导读 在使用conda安装pytorch的时候下载特别慢,有时候下载到一半了还会断掉,然后还需要重新下载才行,很难受. 比如我们要安装pytorch1.3.1的版本,安装命令如下 conda inst ...
- 【conda安装pytorch总是下载cpu版本的问题】
conda安装pytorch总是下载cpu版本的问题 首先,表述问题:我在使用pytorchu官方网站安装torch1.9.0的GPU包, 链接: link 确保自己的安装的代码为 // cuda 1 ...
- 如何解决CentOS下使用yum安装python-pip失败
用Ubuntu的时候感觉很简单的事到ContOS上却变得很头痛,CentOS下使用yum安装python-pip失败的完美解决方法. 在执行以下命令安装Python-pip居然什么也没执行 yum i ...
- CentOS下Redis的安装
CentOS下Redis的安装 前言 安装Redis需要知道自己需要哪个版本,有针对性的安装,比如如果需要redis GEO这个地理集合的特性,那么redis版本就不能低于3.2版本,由于这个特性是3 ...
- Centos下gcc的安装、gcc的更新、gcc安装过程中报错:make[1]: *** [stage1-bubble] 错误 2
Centos下gcc的安装.gcc的更新.gcc安装过程中报错:make[1]: *** [stage1-bubble] 错误 2 文章目录: 1 下载gcc的源码 1.1 源码下载地址: 1.2 下 ...
- CentOS下软件的安装
Linux操作系统上软件的安装从整体上来说有两种方法: 一种是二进制安装,另一种是源码安装 下面我们就先说一下二进制安装 二进制安装我们一般采用两种工具rpm工具和yum工具 1.rpm工具 rpm ...
最新文章
- uboot2014.10移植(一)
- C#多线程学习之(五)使用定时器进行多线程的自动管理
- 二叉树的的前序遍历和后序遍历(题型变种更新中)
- SAP CRM WebClient UI F4 value help is centrally implemented in CRM_THTMLB_UTIL/F4Frameset.html
- 播放2.0:Akka,Rest,Json和依赖项
- php 面试靠快速排序,搞定PHP面试 - 常见排序算法及PHP实现
- php mysql ajax登录界面_PHP+jQuery+Ajax实现用户登录与退出
- 在nodeJs的Express框架下用TypeScript编写router路由出现import关键字错误的解决方案
- PyTorch 和 TensorFlow 引领未来
- Java常用中间件---SpringBoot整合ActivityMQ
- 58同城峰会落幕 智能化平台和下沉市场能让继续神奇吗?
- stm32零基础应该怎么入门?
- php中的oop思想,php oop的特性是什么?php oop思想的三大特点介绍
- 气球java游戏_气球游戏-2019腾讯笔试
- linux 桌面 v2ex,程序员:他人笑我桌面太凌乱,我笑他人看不穿
- 10004 - Bicoloring
- linux服务器上装r,linux服务器安装R语言及Rstudio server
- 第五周学习总结-HTML5
- Excel建立股票资产定价模型,寻找最佳投资组合
- 如何优雅地蹭邻居家的wifi
热门文章
- js 能实现监听F5页面刷新子iframe 而父页面不刷新
- 计算机:游戏14,英语专四作文满分范文(十四):电脑游戏
- android中进行https连接的方式的详解
- sqlserver2005 openRowSet 和row_Number
- celery cluser redis_celery结合redis 使用
- 自走棋投降代码_自走棋太火惨遭针对!代码哥之后又是锁钱挂?玩家:还有拒绝员!...
- web开发时浏览器兼容性问题汇总
- 使用Python在ArcGIS中编程杂谈
- 程序转制作人——游戏设计课程导论
- java中for 的几种常见用法