Hatékonyabb prímkereső basic szinten

import math

szam = int(input("Adjon meg egy számot"))
prim =True

if (szam==1 or szam == 0):
    print (szam, " nem prímszám!\n")
    prim = False
elif(szam%2==0 and szam!=2):
    print (szam, " nem prímszám!\n")
    prim = False
elif(szam==2):
    prim=True
else:
    for i in range(3,round(math.sqrt(szam),None),2):
        if (szam%i==0):
            prim = False
            print(szam," nem primszám!")
            break

if (prim == True):
    print (szam, " primszám!")
input("Nyomjon le egy billentyűt")