当前位置:网站首页>查看 CUDA cudnn 版本 & 测试 cuda 和 cudnn 有效性「建议收藏」
查看 CUDA cudnn 版本 & 测试 cuda 和 cudnn 有效性「建议收藏」
2022-08-10 11:50:00 【全栈程序员站长】
大家好,又见面了,我是你们的朋友全栈君。
https://medium.com/@changrongko/nv-how-to-check-cuda-and-cudnn-version-e05aa21daf6c
cuda 版本 cat /usr/local/cuda/version.txt
cudnn 版本 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
测试 系统 cuda 和 cudnn 有效性
test_cuda.py
import torch
print(2.0)
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
# Assume that we are on a CUDA machine, then this should print a CUDA device:
print(device)
x = torch.Tensor([2.1])
xx = x.cuda()
print(xx)
# CUDNN TEST
from torch.backends import cudnn
print('cudann is ' + str(cudnn.is_acceptable(xx)))
发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/129961.html原文链接:https://javaforall.cn
边栏推荐
- Excel函数公式大全—LOOKUP函数
- 国外媒体宣发怎样做才可以把握重点
- Threshold-based filtering buffer management scheme in a shared buffer packet switch论文核心部分
- 阿里云贾朝辉:云XR平台支持彼真科技呈现国风科幻虚拟演唱会
- An enhanced dynamic packet buffer management. The core part of the paper
- An enhanced dynamic packet buffer management.论文核心部分
- Cannot find symbol log because lombok is not found
- Network Fundamentals (Section 1)
- LeetCode 61. 旋转链表
- 什么是云流化?
猜你喜欢
So delicious!Since using this interface artifact, my team efficiency has increased by 60%!
StarRocks on AWS 回顾 | Data Everywhere 系列活动深圳站圆满结束
If someone asks you about distributed transactions again, throw this to him
郭晶晶家的象棋私教,好家伙是个机器人
你有一份斗破苍穹词库,请查收
Chapter9 : De Novo Molecular Design with Chemical Language Models
人脸考勤是选择人脸比对1:1还是人脸搜索1:N?
2016,还是到了最后
dedecms支持Word内容一键导入
IDC第一的背后,阿里云在打造怎样的一朵“视频云”?
随机推荐
LeetCode 146. LRU 缓存
Does face attendance choose face comparison 1:1 or face search 1:N?
爱可可AI前沿推介(8.10)
【集合】HashSet和ArrayList的查找Contains()时间复杂度
Excel function formulas - HLOOKUP function
LeetCode 146. LRU Cache
Does your child lack self-discipline?Ape Counseling: Pay attention to "blank" in the schedule to give children more control
Database management tool: dynamic read-write separation
如何让别人看不懂你的 JS 代码?把你当大佬!
培训机构学习费用是多少呢?
Excel函数公式大全—HLOOKUP函数
厚积薄发!安全狗再次获得科技成果转化认证!
mpf6_Time Series Data_quandl_correct kernel PCA_AIC_BIC_trend_log_return_seasonal_decompose_sARIMAx_ADFull
Servlet---Solve the problem of Chinese garbled characters in post requests
CLIP还能做分割任务?哥廷根大学提出一个使用文本和图像prompt,能同时作三个分割任务的模型CLIPSeg,榨干CLIP能力...
7、Instant-ngp
Dining (网络流)
LeetCode 24. 两两交换链表中的节点
three.js模糊玻璃效果
【Redis】内存回收策略