Skip to content

Instantly share code, notes, and snippets.

View jacobastern's full-sized avatar

Jacob Stern jacobastern

View GitHub Profile
@sgraaf
sgraaf / ddp_example.py
Last active June 7, 2024 16:26
PyTorch Distributed Data Parallel (DDP) example
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from argparse import ArgumentParser
import torch
import torch.distributed as dist
from torch.nn.parallel import DistributedDataParallel as DDP
from torch.utils.data import DataLoader, Dataset
from torch.utils.data.distributed import DistributedSampler
from transformers import BertForMaskedLM
@nathzi1505
nathzi1505 / install-docker.sh
Last active June 22, 2024 11:56
Docker and Nvidia Docker installation in Ubuntu 20.04 LTS
# WARNING : This gist in the current form is a collection of command examples. Please exercise caution where mentioned.
# Docker
sudo apt-get update
sudo apt-get remove docker docker-engine docker.io
sudo apt install docker.io
sudo systemctl start docker
sudo systemctl enable docker
docker --version
@JoaoRodrigues
JoaoRodrigues / uniprot_rest_example.py
Created June 7, 2019 22:54
Example of Uniprot REST API (Python)
#!/usr/bin/env python
"""
Queries Uniprot database and retrieves (canonical) sequences.
"""
import argparse
import sys
import urllib.request