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