initialize Conv2D named pytorch code example Example: pytorch conda environment conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch