pytorch cpu版本和gpu版本,ubuntu pytorch gpu

　　框架官方网站：

　　https://pytorch.org/

　　复制上述语句，执行，但速度超级超级慢。即便有梯子，速度依旧很慢。容易超时失败！

　　遇到好多好多坑之后，决定采用阿里云的源文件,即在上述代码后附上以下代码：

　　-I http://镜子。阿里云。com/pypi/simple/-trusted-host mirrors . aliyun . com相当于，用阿里云的pip源并且信任这个源，可以用。

　　具体安装如下：（最好管理员身份打开）

　　1.更新点

　　python-mpipinstall-upgrade pip-I http://mirrors . aliyun . com/pypi/simple/-trusted-host mirrors . aliyun . com 2 .安装火炬1.1.0

　　pip安装https://下载。py火炬。org/whl/cu90/torch-1。1 .0-cp37-cp37m-win _ amd64。whl-I http://mirrors.aliyun.com/pypi/simple/-可信主机mirrors.aliyun.com（注：本案例python版本为3.7，其他版本如python3.6，则修改成cp36-cp36m）

　　3.安装火炬视觉

　　pip 3安装https://下载。py火炬。org/whl/cu90/torch vision-03 .0-cp37-cp37m-win _ amd64。whl-I http://mirrors.aliyun.com/pypi/simple/-可信主机mirrors.aliyun.com（注意事项同上）

　　测试torch能否用GPU

　　进口火炬

　　火炬。__版本_ _

　　torch.cuda.is_available()

　　输出真实的即可。

　　图上，表示安装成功啦！

郑重声明：本文由网友发布，不代表盛行IT的观点，版权归原作者所有，仅为传播更多信息之目的，如有侵权请联系，我们将第一时间修改或删除，多谢。