安装Pytorch0.4.0和torchvision-0.2.0(含如何将pip命令的下载源默认成阿里云)


pip install torch==0.4.0 -f https://download.pytorch.org/whl/cu90/stable(按下面参考博客设置好默认阿里云后,才能使此命令加速下载)(可将0.4.0改为所需版本)

参考
https://blog.csdn.net/qq_18649781/article/details/89848266


pip install torchvision==0.2.0(可将0.2.0改为所需版本)

注:
1
torch和torchvision需配套安装,配套关系可查看官网安装命令
查的
例如下面官网命令

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

从中可看出torch1.1.0和torchvision0.3.0配套

2
Ubuntu 下 Pytorch, Tensorflow 对应的Python、英伟达显卡驱动、CUDA、CUDNN版本与环境信息查看方法

你可能感兴趣的:(采坑及软件安装)