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
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