记住了 就这条命令,以pytorch0.4.1,cuda10,torchvison0.2.1为例,可以自由更换版本
conda install pytorch=0.4.1 torchvision=0.2.1 cudatoolkit=10.0 -c pytorch
记住了 就这条命令,以pytorch0.4.1,cuda10,torchvison0.2.1为例,可以自由更换版本
conda install pytorch=0.4.1 torchvision=0.2.1 cudatoolkit=10.0 -c pytorch