window10安装pytorch

# CUDA 10.0
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 9.2
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

一条命令即可安装,其它相应的安装包可以上https://pytorch.org/get-started/previous-versions/进行下载,希望对大家有帮助。 

 

 

你可能感兴趣的:(window10安装pytorch)