Estou tentando instalar uma imagem do docker para pytorch , mas quero usá-la notebook jupyter para codificar pytorch. Como posso editar o código para que eu possa fazer isso?