pytorch安装2020

官方一般只给最新版,如果cuda版本或者驱动版本匹配不上,官方推荐安装的会报错,不能用。

 

pytorch 1.4.0是cuda10.1,装在cuda10.0也能用

pip install https://download.pytorch.org/whl/cu100/torch-1.4.0-cp36-cp36m-win_amd64.whl

历史版本下载

https://blog.csdn.net/jacke121/article/details/86593234

 

cpu:

pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

你可能感兴趣的:(torch)