Install the specified version of the gpu version of pytorch and torchvison under linux

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

 

Guess you like

Origin blog.csdn.net/a493823882/article/details/104775051