henrywang1

# Push subtree in local branch to heroku
git push heroku `git subtree split --prefix project-name local-branch-name`:master --force

henrywang1

from __future__ import print_function
import argparse
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torchvision import datasets, transforms

class Net(nn.Module):
    def __init__(self, share_block):

henrywang1

Install nvidia driver

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-396 nvidia-modprobe
sudo reboot

install CUDA 9.2, cuDNN 7.2.1

copy from: https://gist.github.com/kylemcdonald/3ae0b88a1bf91afc00ba441fe6823a17

henrywang1