当前位置: 首页 > article >正文

centos7配置pytorch和tensorflow

1、安装anaconda

1.1镜像源下载对应anaconda版本后传到服务器上

1.2进入对应文件夹

首先赋权再执行安装程序

chmod +x Anaconda3-2022.10-Linux-x86_64.sh 
./Anaconda3-2022.10-Linux-x86_64.sh chmod +x Anaconda3-2022.10-Linux-x86_64.sh 

1.3交互确认

确认许可协议(一路enter,然后yes)

 更改安装路径/mnt/data/anaconda3

 确认.bashrc文件中初始化Anaconda3的相关内容。yes

 添加内容如下

 安装完成后,直接输入conda应该会提示说没找到该命令,这时候执行:source ~/.bashrc,之后就可以正常使用了。

1.4配置其他用户可用

修改文件vim /etc/profile在最后一行添加

export PATH="$PATH:/mnt/data/anaconda3/bin"

重启生效

source /etc/profile

共享环境

 groupadd anaconda  # 创建anaconda组
 chgrp -R anaconda /mnt/data/anaconda3# 组内共享这个目录下的文件
 chmod 770 -R /mnt/data/anaconda3# 设置权限
 usermod -a -G anaconda username# 添加用户进组
 source /etc/profile # 进组的用户可能需要自己刷新一下

1.5更新默认pip源

# pip安装
python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

1.6更新conda源

修改文件/root/.condarc

channels:
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
  - https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
show_channel_urls: true

2、pytorch安装

查询版本

 目前的驱动和cuda版本最多支持到pytorch1.10.0版本Previous PyTorch Versions | PyTorch

创建环境

conda create -n pytorch python=3.8

更新源

python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

安装pytorch

pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

服务器下载太慢,从本地下好后传上去

cd /mnt/data/anaconda3/envs/pytorch

pip install torch #先别回车,按一下键盘上的Tab键,会自动补全成下面的命令

回车,再次输入

pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0

 安装成功

 python
import torch
torch.__version__

 返回

3、tensorflow安装

安装2.4版本

tensorflow · PyPI下载后传输到服务器文件夹。进入文件夹 

pip install tensorflow-2.4.0-cp38-cp38-manylinux2010_x86_64.whl 

 安装cudatoolkit11.0

conda install cudatoolkit=11.0 -c http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main

安装cudnn8.0 

去nvidia官网下载cuDNN Archive | NVIDIA Developer

上传到对应文件夹,解压压缩包

tar zxvf cudnn-11.0-linux-x64-v8.0.5.39.tgz 

 将解压出的两个文件夹中的文件复制到tf2下的对应文件夹中

 python
import tensorflow as tf
print(tf.__version__)
tf.test.is_gpu_available()

返回True,安装成功


http://www.kler.cn/a/4137.html

相关文章:

  • Vue数据响应式,reaction,ref的使用
  • 在Linux上如何让ollama在GPU上运行模型
  • 【经济学通识——国债】
  • 使用 Java 实现基于 DFA 算法的敏感词检测
  • IO模型与NIO基础二
  • Kafka权威指南(第2版)读书笔记
  • 猿创征文 | re:Invent 朝圣之路:“云“行业风向标
  • 【敏捷开发】jenkins「CI持续集成 CD持续部署」
  • 基于 OpenCV 与 Java 两个语言版本实现获取某一图片特定区域的颜色对比度
  • 从GPT-4、文心一言再到Copilot,AIGC卷出新赛道?
  • Kafka在Mac下的安装与使用
  • 多级评论单表结构设计
  • Maven和Eclipse联合开发
  • js 数据类型
  • 如何将pdf大小压缩?怎么缩小pdf的文件大小?
  • 关于微信小程序安装npm的过程,从下载到小程序内部安装完成
  • gitignore文件的作用和语法
  • 设置鼠标右键打开方式,添加IDEA的打开方式
  • Mac M1通过VMWare Fusion安装Centos7记录(镜像和网络有大坑)
  • 基于OpenCV的图片和视频人脸识别
  • Python实战,爬取金融期货数据
  • 配置案例丨EtherCAT转Profinet网关连接凯福科技总线步进驱动器
  • macOS 13.3(22E252)/12.6.4/11.7.5正式版发布
  • 在芯片设计行业,从项目的初期到交付,不同的岗位的工程师主要负责什么?
  • FairyGUI个人使用手册 只有重点(1)
  • 基于opencv的边缘检测方法