ryujaehun

ryujaehun

devices: (
{
    name: "MX Vertical Advanced Ergonomic Mouse";
    smartshift:
    {
        on: true;
        threshold: 30;
    };
    hiresscroll:
    {

ryujaehun

import torch
import torchvision.models as models
# import argparse
# parser = argparse.ArgumentParser()
# parser.add_argument("-n", "--network", help="network for onnx file", required=False)
# args = parser.parse_args()
input_size = (1,3,224,224)

dummy_input = torch.randn(*input_size, device="cuda")
MODEL_LIST = {

ryujaehun

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#

ryujaehun

tar -xvf cudnn-*.tar.xz
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*