![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
cuda
RichardorMu
四处搬砖,HCI,gaze estimation,music,jetson
展开
-
Jetson 系列——[LibTorch & Linux] 各版本 LibTorch 下载
以下全为 Linux 版本。Win 版本单击 这里(https://blog.csdn.net/weixin_43742643/article/details/115218126)所有版本都是已经编译好的,解压后就可以使用。libtorch 的版本和 pytorch 是对应的,比如 libtorch 1.6.0 对应于 pytorch 1.6.0。cuda 的兼容性问题新版本无法在旧机器上运行。比如 libtorch 1.6.0 的 cu102 版本无法在 cuda-9.0 的机器上运行(已实测)原创 2021-08-27 15:17:44 · 1136 阅读 · 1 评论 -
Jetson 系列——Jetson Nano使用sudo命令免输入密码方法
1、打开终端,输入命令:sudo visudo2、在文档最后一行添加如下内容:xxx ALL=(ALL)NOPASSWD:ALL保存退出即可其中XXX为登入用户名。原创 2021-08-27 09:40:48 · 1853 阅读 · 1 评论 -
pytorch的显存释放机制torch.cuda.empty_cache()
Pytorch已经可以自动回收我们不用的显存,类似于python的引用机制,当某一内存内的数据不再有任何变量引用时,这部分的内存便会被释放。但有一点需要注意,当我们有一部分显存不再使用的时候,这部分释放的显存通过Nvidia-smi命令是看不到的,举个例子:device = torch.device('cuda:0')# 定义两个tensordummy_tensor_4 = torch.randn(120, 3, 512, 512).float().to(device) # 120*3*512*51原创 2021-07-10 17:06:32 · 2036 阅读 · 0 评论 -
查看linux上cuda版本
查看 CUDA 版本:cat /usr/local/cuda/version.txt查看 CUDNN 版本:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2原创 2019-10-24 14:44:00 · 1657 阅读 · 0 评论