在Ubuntu下安装PyTorch（GPU版）的简易教程 - 办公管理系统

热门文章排行

推荐文章

关键词

PyTorch 安装

在Ubuntu下安装PyTorch（GPU版）的简易教程

准备工作

1.需要确保你的电脑上安装了CUDA，并且CUDA的版本支持PyTorch，可以参考PyTorch官网上提供的CUDA版本对照表来查看。
2.你需要确定你的电脑上是否安装了PyTorch的依赖库，比如说C++编译器、Python环境等，如果没有安装，可以参考PyTorch官网上提供的安装指南。

安装PyTorch

1.打开终端，输入以下命令来安装PyTorch：
```
$ pip install torch torchvision
```

2.根据你的CUDA版本，输入以下命令来安装PyTorch：

# 如果CUDA的版本是9.0
$ pip install torch torchvision cuda90

# 如果CUDA的版本是9.2
$ pip install torch torchvision cuda92

# 如果CUDA的版本是10.0
$ pip install torch torchvision cuda100

# 如果CUDA的版本是10.1
$ pip install torch torchvision cuda101

# 如果CUDA的版本是10.2
$ pip install torch torchvision cuda102

# 如果CUDA的版本是11.0
$ pip install torch torchvision cuda110

3.输入以下命令来检查是否安装成功：
```
$ python
>>> import torch
>>> print(torch.__version__)
```
如果能够正确输出PyTorch的版本号，表明安装成功。

本文链接：http://task.lmcjl.com/news/2564.html

展开阅读全文

上一篇：Python Pandas中的factorize()函数解析：数据标签化方法下一篇：Linux命令详解：history显示用户命令历史记录

▲