Remember this command, take pytorch0.4.1, cuda10, torchvison0.2.1 as examples, you can change the version freely
conda install pytorch=0.4.1 torchvision=0.2.1 cudatoolkit=10.0 -c pytorch
Remember this command, take pytorch0.4.1, cuda10, torchvison0.2.1 as examples, you can change the version freely
conda install pytorch=0.4.1 torchvision=0.2.1 cudatoolkit=10.0 -c pytorch