window10安装pytorch

1、nvcc -V 查看cuda版本

 2、查看pytorch版本

 3、安装pytorch

pip install torch==1.0.1 -f https://download.pytorch.org/whl/torch_stable.html

 4、安装torchvision

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ torchvision==0.2.2

原文地址:https://www.cnblogs.com/wjjcjj/p/13786925.html