pytorch的GPU版本安装_Novelin的博客-程序员秘密

首先查看自己的 cuda 版本

在 NVIDIA 控制面板，左下角系统信息，组件，可以看到 NVCUDA.DLL 的版本号

然后在这里找到对应版本

pytorch_gpu
然后在命令中加入 --default-timeout=2000，下载过程中会有一段时间比较慢

pip install --default-timeout=2000 torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

然后下载完，测试一下

import torch
x = torch.rand(5, 3)
print(x)

import torch
torch.cuda.is_available()

如果返回 True 就好啦

本文链接：https://blog.csdn.net/qq_40860934/article/details/111125087

原作者删帖不实内容删帖广告或垃圾文章投诉

智能推荐

AD绘制PCB经验总结3-规则报警 Silk primitive without silk layer+Minimum Solder Mask Sliver_苍茫也初心的博客-程序员秘密

PCB画完进行规则检查时，总会出现各种各样的问题，有些问题莫名其妙不知其然。下面就我遇到的几个，记录下。1.Silk primitive without silk layer，意思就是有些丝印超出了丝印该在的区域。但是一般到规则检查时，丝印都摆放整齐了，不应该出现丝印在外面的情况啊，这是为何呢？一个元器件有好多个属性，一般我们把元器件的标号作为显示的丝印，而元器件的值却隐层起来，这导致隐层的丝印有可能跑到外面去，所以就报错了。方法呢，我们可以忽略该问题，要不就把隐藏的丝印摆放好就ok了。或者把上

用python进行渗透测试_利用Python进行Web渗透测试（五）：剖析HTTP请求_weixin_39679718的博客-程序员秘密

本篇概要URL的结构HTTP头域HTTP状态码一个完整的HTTPget请求在上一篇我们简单介绍了HTTP协议的概念和知识，本篇我们来详细剖析一下HTTP的请求，将这两篇综合起来，我们就对HTTP有了基本的认识和了解了。URL结构解析通常情况下，我们发起一个HTTP请求都是通过访问一个URL网址来实现。比如我们打开百度的首页(http://www.baidu.com)，就是对百度的服务器发起了一个H...

基于Node.js http-proxy库的本地HTTP代理服务器脚本_http-proxy支持rewrite么_志_祥的博客-程序员秘密

此脚本的主要目的：让浏览器客户端首先通过此代理，然后再访问目标网站。可基于特定的url做正则表达式匹配，然后替换为本地修改的版本。主要用于浏览器内核研发过程中某些怀疑是特定网站JS脚本执行出错导致的渲染异常bug。可在JS脚本中插入console.log以定位出错点。var ProxyPort = 8888;var http = require('http'), net

MT6732芯片资料MT6732芯片最新参考设计MT6732芯片规格书_普通网友的博客-程序员秘密

MT6732Platform‐ 基础介绍LTE4G（64bit）4核芯片- 8926 与高通芯片性能相当, BOMcost贴近目前6582主流价格,1.5GHz主频，HD分辨率屏，13MP摄像头；与MT67...

MongoDB 副本集的常用操作及原理_weixin_30329623的博客-程序员秘密

本文是对MongoDB副本集常用操作的一个汇总，同时也穿插着介绍了操作背后的原理及注意点。结合之前的文章：MongoDB副本集的搭建，大家可以在较短的时间内熟悉MongoDB的搭建和管理。下面的操作主要分为两个部分：1. 修改节点状态主要包括： 1>将Primary节点降级为Secondary节点 2>冻结Secondary节点...

机器学习算法——决策树_决策树splitter_青稣儿的博客-程序员秘密

1.什么是决策树？决策树是一种常见的分类模型，基于树结构对数据进行划分。广泛应用于金融分控、医疗辅助诊断等诸多行业。决策树的步骤：2.决策树的构建决策树的构建过程是一个递归过程。伪代码如下：2.1信息增益...