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

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

  输出真实的即可。

  图上,表示安装成功啦!

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