当前位置：网站首页>深度学习------pytorch-gpu环境搭建
深度学习------pytorch-gpu环境搭建
2022-08-04 11:21:00 【小飞龙程序员】
本人笔记本1650显卡，pytorch（gpu）环境搭建
cuda10.2gpu安装命令：pip install torch1.10.1+cu102 torchvision0.11.2+cu102 torchaudio0.12.1 -f https://download.pytorch.org/whl/torch_stable.html
但是，torch和torchvision版本必须匹配
pip install torch1.10.1+cu102 torchvision==0.11.2+cu102 -f https://download.pytorch.org/whl/torch_stable.html
验证cuda与gpu是否连接成功？
import torch
DEVICE = ("cuda" 边栏推荐
- 3-5年以上的功能测试如何进阶自动化？
- The use of DDR3 (Naive) in Xilinx VIVADO (1) to create an IP core
- Learn to use the basic interface of set and map
- 命令模式（Command）
- win8和win10下，visual studio 2008 调试出现无响应的卡死问题解决
- 萌宠来袭，如何让“吸猫撸狗”更有保障？
- 手搓一个“七夕限定”，用3D Engine 5分钟实现烟花绽放效果
- SkiaSharp 之 WPF 自绘 粒子花园（案例版）
- 小程序实战（一）- 骨架屏的应用与实现
- 超美星空特效，你Get了吗？
猜你喜欢
Leetcode刷题——二叉搜索树相关题目（98. 验证二叉搜索树、235. 二叉搜索树的最近公共祖先、1038. 从二叉搜索树到更大和树、538. 把二叉搜索树转换为累加树）
What is the principle of thermal imaging temperature measurement?Do you know?
热成像测温的原理是什么呢？你知道吗？
Leetcode - using sequence traversal features first completed 114. The binary tree to the list
Xilinx VIVADO 中 DDR3（Naive）的使用（2）读写设计
The use of DDR3 (Naive) in Xilinx VIVADO (2) Read and write design
【目标检测】yolov3特征提取网络------Darknet53网络及pytorch实现
表的完整性约束;非外键约束
Advanced transcriptome analysis and R data visualization hot registration (2022.10)
MySql数据库入门的基本操作
随机推荐
数据库对象-视图；存储过程
网管交换机与非网管交换机如何选择？
【LeetCode】653. 两数之和 IV - 输入 BST
数据库对象
Camunda overall architecture and related concepts
【机器学习】：如何对你的数据进行分类？
Xilinx VIVADO 中 DDR3（Naive）的使用（3）仿真测试
强烈推荐一款优秀且通用的后台管理系统
上帝空间——全球首个基于Web3.0的艺术协议创意平台，拓宽多元艺术融合边界
Mysql——》类型转换符binary
MySQL不提供数组，只能做成表吗？
超美星空特效，你Get了吗？
你值得拥有的登录注册页面（附赠源码）
网管型交换机比傻瓜交换机多了哪些功能
BOSS 直聘回应女大学生连遭两次性骚扰：高度重视求职者安全，可通过 App 等举报
*W3C* 标准组织
云原生Devops 的实现方法
Leetcode brush - structure binary tree (105. Once upon a time sequence and the sequence structure binary tree traversal sequence, 106. From the sequence with the sequence structure binary tree travers
Leetcode Brush Questions - Path Sum
多表查询