您当前的位置:首页 > IT编程 > python
| C语言 | Java | VB | VC | python | Android | TensorFlow | C++ | oracle | 学术与代码 | cnn卷积神经网络 | gnn | 图像修复 | Keras | 数据集 | Neo4j | 自然语言处理 | 深度学习 | 医学CAD | 医学影像 | 超参数 | pointnet | pytorch | 异常检测 | Transformers | 情感分类 | 知识图谱 |

自学教程:安装GPU版本pytorch教程

51自学网 2021-11-13 21:57:55
  python
这篇教程安装GPU版本pytorch教程写得很实用,希望能帮到您。

安装GPU版本pytorch教程(避坑)

 

Pytorch官方网站:

https://pytorch.org/

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

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

-i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

相当于,用阿里云的pip源并且信任这个源,可以用。

 

具体安装如下:(最好管理员身份打开)

1.更新Pip

python -m pip install --upgrade pip -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

2.安装torch 1.1.0

pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl  -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

(注:本案例python版本为3.7,其他版本如python3.6,则修改成cp36-cp36m)

3.安装torchvision 0.3.0

pip3 install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl  -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

(注意事项同上)

 

测试torch能否用GPU

>>>import torch

>>>torch.__version__

>>>torch.cuda.is_available()

输出True即可。

图上,表示安装成功啦!!!

 

 

 
  • 22
    点赞
  • 13
    评论
  • 99
    收藏
  • 一键三连
    一键三连
  •  
  • 扫一扫,分享海报

     
表情包
插入表情
 
 
 
©️2021 CSDN 皮肤主题: 终极编程指南 设计师:CSDN官方博客 返回首页

Python批量合并相同格式的Excel文件
解决报错ImportError: unique_cuda.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN6caffe28T
万事OK自学网:51自学网_软件自学网_CAD自学网自学excel、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。