install torch.nn code example

Example 1: pip install pytorch windows

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

Example 2: PyTorch

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch