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

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.lryc.cn/news/43633.html

相关文章:

  • Kafka在Mac下的安装与使用
  • AndroidStudio相对布局
  • 如何用iOS自带摄像头进行拍摄获取视频流以及OpenCV图像处理实时显示
  • 智安网络|为什么说防火墙是我们信息安全的第一道防线?
  • Android多媒体功能开发(8)——使用VideoView控件播放视频
  • python调用CC++
  • [golang gin框架] 10.Gin 商城项目介绍
  • Endor Labs:2023年十大开源安全风险
  • 关于Error和Exception的一些思考 小结
  • Mac环境变量配置(Java)
  • 通过这三个文件彻底搞懂rocketmq的存储原理
  • Linux安装Nvidia显卡驱动
  • GPT-4 介绍
  • Ubuntu下单机安装Hadoop详细教程(附所需安装包下载)
  • 【嵌入式烧录/刷写文件】-2.1-详解Intel Hex格式文件
  • 【云原生】初识 Kubernetes — pod 的前世今生
  • 【基础篇】Java类加载器详解
  • Pytorch动手实现Transformer机器翻译
  • 宝塔面板部署node+vue项目注意事项
  • 【LeetCode】剑指 Offer 39. 数组中出现次数超过一半的数字 p205 -- Java Version
  • fisco bcos用caliper0.2.0进行压力测试的安装配置
  • 正在进行 | 用友企业数智化财务峰会落地广州 高能不断
  • uniapp - APP云打包、蒲公英平台发布APP的步骤
  • reposync命令详解--reposync同步aliyunyum库到本地
  • OCR之论文笔记TrOCR
  • 雷电4模拟器安装xposed框架(2022年)
  • 微信小程序支付完整流程(前端)
  • 设置鼠标右键打开方式,添加IDEA的打开方式
  • LAMP架构之zabbix监控(2):zabbix基础操作
  • ShareSDK常见问题