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

pytorch各种版本最简单安装,不用自己安装cuda cudnn

pytorch各种版本 pip 安装命令 查看官网
https://pytorch.org/get-started/previous-versions/
在这里插入图片描述

conda +pytorch 安装

1、安装conda,
2、创建并并激活虚拟环境
- conda create -n pytorch_1.7 python=3.7
- conda activate pytorch_1.7
3、虚拟环境中 pip 安装想要的版本  不用先把cuda cudnn 先安装了,如下会自动安装好3.1  pip 命令安装(有时候下载超时不行用3.2离线安装方式)复制pip  命令pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html4、查看pytorch 版本和对应cuda cudnn 版本是不是正确
python
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())

docker + paddle + pytorch

注意 paddle 和 pytorch cuda版本支持一致

1、paddle 官网下载 打包好的镜像
https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/linux-pip.html
在这里插入图片描述

2、安装流程

Paddle docker 2.3.2官网镜像容器安装(最新的按照官方命令替换)
(1)下载镜像:
nvidia-docker pull registry.baidubce.com/paddlepaddle/paddle:2.3.2-gpu-cuda11.2-cudnn8.2-trt8.0
(2)建立容器

  • nvidia-docker run -it --name paddle_2.3.2 --gpus all --shm-size 32g -v /home/syy:/home/syy paddlepaddle/paddle:2.3.2-gpu-cuda11.2-cudnn8 /bin/bash

(3)进入容器
docker exec -it paddle_2.3.2 /bin/bash

(4)安装pytorch (找cuda 11.2的包 pip,)
pytorch 官网没有cuda11.2的包,所以不能安装,官方有cuda11.3 11.6,所以想兼容到一个环境上述paddle 要下载 cuda11.6的镜像

http://www.lryc.cn/news/9434.html

相关文章:

  • 订单超时处理方案介绍
  • Blackbox-Exporter对服务进行探活
  • react-redux
  • 算法刷刷刷| 回溯篇| 子集问题大集合
  • 合并两个有序数组-力扣88-java
  • 2022「大厂可观测」重磅回顾,12场直播,15位技术大咖洞见可观测
  • CMMI-配置管理(CM)
  • 网络编程套接字Socket
  • Linux进程概念(二)
  • 墨天轮【第二届数据库掌门人论坛】圆满收官 | 含嘉宾精彩观点回顾
  • Redis之集群搭建
  • 31-Golang中的二维数组
  • <<Java开发环境配置>>6-SQLyog安装教程
  • MySQL 中的 distinct 和 group by 哪个效率更高
  • 计算机相关专业毕业论文选题推荐
  • 网络编程套接字之TCP
  • 网络与串口调试工具TCPCOM
  • 数据库常用命令
  • PTA复习
  • TypeScript 学习之接口
  • 原码反码补码
  • 大数据选股智能推荐系统V1.0-1
  • 调研生成GIF表情包之路
  • 【RocketMQ】源码详解:生产者启动与消息发送流程
  • 信息安全(一)
  • 企业多会场视频直播(主会场、分会场直播)实例效果
  • 线性代数速览(一)行列式
  • 恭喜山东翰林“智慧园区管理系统”获易知微可视化设计大赛二等奖
  • gulp简单使用
  • ce认证机构如何选择?