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()
输出真实的即可。
图上,表示安装成功啦!