当前位置:网站首页>查看 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
边栏推荐
- CLIP还能做分割任务?哥廷根大学提出一个使用文本和图像prompt,能同时作三个分割任务的模型CLIPSeg,榨干CLIP能力...
- jlink and swd interface definition
- You have a Doubaqiong thesaurus, please check it
- Dining (web stream)
- LeetCode 138. 复制带随机指针的链表
- Configure druid data source "recommended collection"
- The 6th "Blue Hat Cup" National College Student Network Security Skills Competition Semi-Final Part WriteUp
- 【集合】HashSet和ArrayList的查找Contains()时间复杂度
- LeetCode 445. Adding Two Numbers II
- 托米的咒语
猜你喜欢
16. Getting Started with Pytorch Lightning
StarRocks on AWS Review | Data Everywhere Series Event Shenzhen Station ended successfully
dedecms支持Word内容一键导入
dedecms supports one-click import of Word content
2016,还是到了最后
Does face attendance choose face comparison 1:1 or face search 1:N?
mpf6_Time Series Data_quandl_更正kernel PCA_AIC_BIC_trend_log_return_seasonal_decompose_sARIMAx_ADFull
StarRocks on AWS 回顾 | Data Everywhere 系列活动深圳站圆满结束
太香了!自从用了这款接口神器,我的团队效率提升了 60%!
7、Instant-ngp
随机推荐
【Redis】内存回收策略
加密游戏:游戏的未来
你有一份斗破苍穹词库,请查收
LeetCode 21. Merge two ordered linked lists
Excel function formulas - HLOOKUP function
Network Fundamentals (Section 1)
Can CLIP also do segmentation tasks?The University of Göttingen proposed a model CLIPSeg that uses text and image prompts to perform three segmentation tasks at the same time, draining CLIP capabiliti
CodeForces - 628D (digital dp)
LeetCode 24. Swap nodes in linked list pairwise
LeetCode 19. 删除链表的倒数第 N 个结点
孩子自律性不够?猿辅导:计划表要注意“留白”给孩子更多掌控感
Golang分布式应用之etcd
bat脚本——提取多个文件夹到指定路径
LeetCode 86. 分隔链表
制品库是什么?
中芯CIM国产化项目暂停?上扬软件:未停摆,改为远程开发!
开源的作者,也有个生活问题
Servlet---Solve the problem of Chinese garbled characters in post requests
因为找不到lombok而找不到符号log
HDU 4135: Co-prime (the principle of inclusion and exclusion)