app.py

import sqlite3
import streamlit as st
import pandas as pd
import natsort
# import locale
# locale.setlocale(locale.LC_ALL, 'en_US.UTF-8')

conn = sqlite3.connect('version.db')

st.set_page_config(
    page_title='PyTorch (on Linux64) Installation Environment Selection',
    page_icon='🖐'
)
pytorch_versions = ['null',
                    '1.5.1',
                    '1.6.0', '1.7.0', '1.7.1', '1.8.0', '1.8.1', '1.9.0', '1.9.1', '1.10.0', '1.10.1', '1.10.2',
                    '1.11.0', '1.12.0', '1.12.1', '1.13.0', '1.13.1', '2.0.0', '2.0.1', '2.1.0', '2.1.1', '2.1.2', '2.2.0', '2.2.1', '2.2.2', '2.3.0', '2.3.1']
python_versions = ['null', '3.6', '3.7', '3.8', '3.9', '3.10', '3.11', '3.12']
cuda_versions = ['null', '9.2', '10.1', '10.2', '11.0',
                 '11.1', '11.3', '11.5', '11.6', '11.7', '11.8', '12.1']
sm_versions = ['null', 'sm_35', 'sm_37', 'sm_50',
               'sm_60', 'sm_61', 'sm_70', 'sm_75', 'sm_80', 'sm_86', 'sm_90']
st.markdown(
    '### Please select the software version and/or compute capability of your NVIDIA GPU (On Linux64)')
st.info('The environment settings are collected from https://conda.anaconda.org/pytorch/linux-64/ \
        and https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#major-components, \
        so other settings may also work.', icon="👋")
st.info('If the driver version contains `**`, \
        `**` means: CUDA 11.0 was released with an earlier driver version, \
        but by upgrading to Tesla Recommended Drivers 450.80.02 (Linux) / 452.39 (Windows), \
        minor version compatibility is possible across the CUDA 11.x family of toolkits.', icon='👇')
col1, col2, col3, col4 = st.columns(4)
with col1:
    option_pytorch = st.selectbox(
        label='PyTorch Version', options=pytorch_versions)
with col2:
    option_python = st.selectbox(
        label='Python Version', options=python_versions)
with col3:
    option_cuda = st.selectbox(
        label='CUDAToolkit Version', options=cuda_versions)
with col4:
    option_sm = st.selectbox(label='SM Version', options=sm_versions)
st.markdown('`null` means no contraint on corresponding option.')


def main():
    if option_pytorch not in pytorch_versions or option_cuda not in cuda_versions or option_python not in python_versions or option_sm not in sm_versions:
        st.error('Please enter valid version')
        return

    cur = conn.cursor()

    sql = 'SELECT * FROM pytorch'
    data = []
    cond = ''
    if option_pytorch != 'null':
        cond += f' AND pytorch = ?'
        data.append(option_pytorch)
    if option_python != 'null':
        cond += f' AND python = ?'
        data.append(option_python)
    if option_cuda != 'null':
        cond += f' AND cuda like ?'
        data.append(option_cuda+'%')
    if option_sm != 'null':
        cond += f' AND sm = ?'
        data.append(option_sm[3:])
    if cond != '':
        sql = sql + ' WHERE ' + cond[4:]
    cur = cur.execute(sql, data)

    df = pd.DataFrame(cur.fetchall(), columns=[
                      'PyTorch', 'Python', 'CUDA', 'cuDNN', 'SM', 'driver>='])
    df = df.sort_values(by=['PyTorch', 'Python', 'CUDA', 'cuDNN', 'SM', 'driver>='],
                        ignore_index=True, key=natsort.natsort_keygen(alg=natsort.ns.LOCALEALPHA))
    st.table(df)


main()