diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-08-25 23:16:25 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-08-25 23:16:25 +0200 |
commit | 54d8b230eedfdf587e2d2d214d65582fe78c47eb (patch) | |
tree | 2516b415bd3605b1164d85192ed943f2f99b38e1 /notebooks/Untitled.ipynb | |
parent | 7c64b48c07d9c39c18819ead31dc95beba012283 (diff) |
Update notebooks
Diffstat (limited to 'notebooks/Untitled.ipynb')
-rw-r--r-- | notebooks/Untitled.ipynb | 276 |
1 files changed, 276 insertions, 0 deletions
diff --git a/notebooks/Untitled.ipynb b/notebooks/Untitled.ipynb new file mode 100644 index 0000000..7ea06ae --- /dev/null +++ b/notebooks/Untitled.ipynb @@ -0,0 +1,276 @@ +{ + "cells": [ + { + "cell_type": "code", + "execution_count": 1, + "id": "8468e45a", + "metadata": { + "tags": [] + }, + "outputs": [], + "source": [ + "import torch" + ] + }, + { + "cell_type": "code", + "execution_count": 2, + "id": "d1bc956b", + "metadata": { + "tags": [] + }, + "outputs": [ + { + "data": { + "text/plain": [ + "True" + ] + }, + "execution_count": 2, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "torch.cuda.is_available()" + ] + }, + { + "cell_type": "code", + "execution_count": 3, + "id": "652cfb26", + "metadata": { + "tags": [] + }, + "outputs": [ + { + "data": { + "text/plain": [ + "1" + ] + }, + "execution_count": 3, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "torch.cuda.device_count()" + ] + }, + { + "cell_type": "code", + "execution_count": 4, + "id": "0fc5e328", + "metadata": { + "tags": [] + }, + "outputs": [ + { + "data": { + "text/plain": [ + "0" + ] + }, + "execution_count": 4, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "torch.cuda.current_device()" + ] + }, + { + "cell_type": "code", + "execution_count": 5, + "id": "4ed93c82", + "metadata": { + "tags": [] + }, + "outputs": [ + { + "data": { + "text/plain": [ + "<torch.cuda.device at 0x7fbf7bf7b9a0>" + ] + }, + "execution_count": 5, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "torch.cuda.device(0)" + ] + }, + { + "cell_type": "code", + "execution_count": 6, + "id": "c03841ce", + "metadata": { + "tags": [] + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'NVIDIA GeForce RTX 2070'" + ] + }, + "execution_count": 6, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "torch.cuda.get_device_name(0)" + ] + }, + { + "cell_type": "code", + "execution_count": 7, + "id": "2fba9680-0c73-4a62-b24d-ceea79874717", + "metadata": { + "tags": [] + }, + "outputs": [ + { + "data": { + "text/plain": [ + "'11.7'" + ] + }, + "execution_count": 7, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "torch.version.cuda" + ] + }, + { + "cell_type": "code", + "execution_count": 10, + "id": "7e62de10", + "metadata": {}, + "outputs": [], + "source": [ + "t = torch.randn((2, 1, 3, 64))" + ] + }, + { + "cell_type": "code", + "execution_count": 11, + "id": "fc221b4e", + "metadata": {}, + "outputs": [ + { + "data": { + "text/plain": [ + "torch.Size([2, 1, 3, 64])" + ] + }, + "execution_count": 11, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "t.shape" + ] + }, + { + "cell_type": "code", + "execution_count": 12, + "id": "ab80d75e", + "metadata": {}, + "outputs": [], + "source": [ + "import torch.nn.functional as F" + ] + }, + { + "cell_type": "code", + "execution_count": 21, + "id": "bfe1fc90", + "metadata": {}, + "outputs": [ + { + "data": { + "text/plain": [ + "torch.Size([2, 1, 7, 128])" + ] + }, + "execution_count": 21, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "F.interpolate(t, scale_factor=[2.5, 2], mode=\"nearest\").shape" + ] + }, + { + "cell_type": "code", + "execution_count": 18, + "id": "f69fa1fc", + "metadata": {}, + "outputs": [], + "source": [ + "f = torch.nn.Conv2d(1, 1, 3, 1, padding=1)" + ] + }, + { + "cell_type": "code", + "execution_count": 20, + "id": "cd6df204", + "metadata": {}, + "outputs": [ + { + "data": { + "text/plain": [ + "torch.Size([2, 1, 64, 64])" + ] + }, + "execution_count": 20, + "metadata": {}, + "output_type": "execute_result" + } + ], + "source": [ + "f(torch.randn(2, 1, 64, 64)).shape" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "id": "41693566", + "metadata": {}, + "outputs": [], + "source": [] + } + ], + "metadata": { + "kernelspec": { + "display_name": "Python 3 (ipykernel)", + "language": "python", + "name": "python3" + }, + "language_info": { + "codemirror_mode": { + "name": "ipython", + "version": 3 + }, + "file_extension": ".py", + "mimetype": "text/x-python", + "name": "python", + "nbconvert_exporter": "python", + "pygments_lexer": "ipython3", + "version": "3.9.4" + } + }, + "nbformat": 4, + "nbformat_minor": 5 +} |