Cnn Model Python Pytorch Install