centos下通过conda安装pytorch

2023-05-16

一、安装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

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

centos下通过conda安装pytorch 的相关文章

随机推荐

  • 机器人ROS系统学习随笔->2 RPLIDAR激光雷达使用

    一 驱动的安装 SLAMTEC官网下载激光雷达资料 xff1a http www slamtec com 本人用的是RPLIDAR A1的激光雷达 在下载界面下载相应的资料 xff0c sdk与固件及其他 二 安装 1 建立工作空间并编译
  • 蓝牙飞控数传套装(适合APM/Pixhaw/Pixhack/Pixhawk2飞控)

    SSC FK BL900是迅瞻电子初创的一款蓝牙 43 射频模块一体的飞控数传套装电台 xff0c 它内嵌入Digi的900HP模块和蓝牙模块 xff0c 对外提供XT60的电池供电接口 xff0c 支持5 28V的宽电压工作 xff0c
  • VC++ 编译过程

    一 前言 一开始编译C 43 43 代码的时候可能会对编译的错误觉得很难理解 xff0c 搞不清楚究竟是哪里错了 了解编译过程 xff0c 能够更好的处理编译错误 二 名词解释 编译单元 xff1a 当一个c或cpp文件在编译时 xff0c
  • PLC的ST编程方式--文本编程,简洁啊

    一 ST语言介绍 发现网上PLC的ST编程资料极少 不过 xff0c 道理也很简单 xff0c 因为做PLC的基本都是电气出身 xff0c 梯形图类似于继电器逻辑 xff0c 比较接近他们的习惯 ST属于文本编程 xff0c 符合程序员的习
  • stm32气压传感器 带探头的_基于STM32的真空度的测量装置的设计

    阮敬华 43 张忠伟 43 徐沛 43 李雪莲 43 阚茹男 摘 要 xff1a 采用USART串口通信 SPI通信作为核心技术 xff0c 通过气压和温湿度传感器对环境的真空值 温湿度参数进行数据采集 xff0c 将采集到的数据通过SD卡
  • Python 迭代器

    一 迭代器 迭代是访问集合元素的一种方式 迭代器是一个可以记住遍历的位置的对象 迭代器对象从集合的第一个元素开始访问 xff0c 直到所有的元素被访问完结束 迭代器只能往前不会后退 1 1 判断一个对象是否可迭代 可以使用 isinstan
  • EasyPlayerPro Windows播放器进行本地对讲喊话音频采集功能实现

    需求 在安防行业应用中 xff0c 除了在本地看到摄像机的视频和进行音频监听外 xff0c 还有一个重要的功能 xff0c 那就是对讲 EasyPlayerPro win为了减轻二次开发者的工作量 xff0c 将本地音频采集也进行了集成 x
  • 大端法、小端法、网络字节序

    关于字节序 大端法 小端法 的定义 UNXI网络编程 定义 xff1a 术语 小端 和 大端 表示多字节值的哪一端 小端或大端 存储在该值的起始地址 小端存在起始地址 xff0c 即是小端字节序 xff1b 大端存在起始地址 xff0c 即
  • 阿里云安全肖力:云原生安全构筑下一代企业安全架构

    34 数字经济的发展驱动越来越多的企业上云 xff0c 每个企业都会基于云原生安全能力构筑下一代企业安全架构 xff0c 完成从扁平到立体式架构的进化 xff0c 届时云原生安全技术红利也将加速释放 xff01 9月27日 xff0c 阿里
  • Vue 自定义按键修饰符

    如点击F2 触发某个事件 lt input type 61 34 button 34 name 61 34 34 id 61 34 34 value 61 34 添加 34 64 keyup f2 61 34 add 34 gt 自定义全局
  • android Studio keytool' 不是内部或外部命令,也不是可运行的程序 或批处理文件

    android Studio keytool 39 不是内部或外部命令 xff0c 也不是可运行的程序 或批处理文件 遇到这个问题好久了 xff0c 一直没解决今天搜集了大量的资料 xff0c 有的说什么Java没配置好 xff0c 不是扯
  • java -jar 运行springboot项目时内存设置

    java Xms64m JVM启动时的初始堆大小 Xmx128m 最大堆大小 Xmn64m 年轻代的大小 xff0c 其余的空间是老年代 XX MaxMetaspaceSize 61 128m XX CompressedClassSpace
  • Jupyter notebook 读取文件的问题

    Jupyter notebook只能打开当前目录下的数据集 xff08 txt CSV等 xff09 xff0c 所以需要把数据集倒导入到当前目录下 xff0c 导入的方法是 1 文件不大时 直接上传文件 pd read csv读取 2 文
  • MIUI目前为止最简单安装谷歌服务框架教程

    安装谷歌服务框架方法有很多 xff0c 比如用第三方 rec卡刷gapps包 用第三方工具安装 然而这些对于新手来说还是比较难的 xff01 我今天说的方法可以说是最简单的 xff1a 1 不需要修改文件 xff1b 2 不需要借助第三方软
  • 用docker安装emby的详细步骤

    搭建个人存储服务器NAS xff0c 媒体播放器少不了 群晖自带的Video Station据说没有Emby Plex好用 Plex是收费的 xff0c Emby是开源的 本文看似很长 实际上有一半篇幅是关于设置 亲测 xff0c 那些设置
  • ubuntu 下通过ftp命令下载文件

    连接 ftp 192 168 180 2 Connected to 192 168 180 2 Name 192 168 180 2 rivsidn admin Password 获取远端文件 ftp gt get test pdf loc
  • pycharm安装到32位操作系统

    在32位操作系统中安装pycharm过程中发现的一些问题 首先是下载了最新版本的pycharm安装后打开 xff0c 弹出未发现可执行的文件 xff0c 然后想到了其他的办法 1 下载最新版本的pycharm不能直接运行 xff0c 因此可
  • 遗忘的初境

    农历八月深夜的黎明朦胧昏黄 xff0c 林立在山雾中的坟地貌似并不太平 xff0c 白露轻风清不走香纸白烛烟气 这层孽障将要带着旧尘怨气破土而出 筹谋着一场突击 这支躲藏在大山幽月中的乡村透着光亮 一双双眼睛凝视着挂在横梁上昏暗的煤灯 泛白
  • 2019-2020-1 20175313 《信息安全系统设计基础》第二周学习总结

    目录 一 教材学习内容总结二 教材学习中的问题和解决过程三 心得体会四 学习进度条五 参考资料 一 教材学习内容总结 第二章内容 基本知识 重点和难点 二 教材学习中的问题和解决过程 问题1 xff1a 对教材49页上的代码进行验证时 xf
  • centos下通过conda安装pytorch

    一 安装anaconda anaconda安装简单 xff0c 只要确定自己的系统即可 xff0c 具体安装请参考这里 二 确定自己的系统版本 我的是centos cat etc redhat release 查看linux系统方法 xff