ubuntu18.04 pytorch安装
时间: 2023-04-25 09:04:21 浏览: 292
ubuntu 16.04 pytorch安装总结
要在Ubuntu 18.04上安装PyTorch,可以按照以下步骤进行操作:
1. 安装Anaconda或Miniconda
2. 创建一个新的conda环境
3. 在新环境中安装PyTorch
具体步骤如下:
1. 安装Anaconda或Miniconda
可以从Anaconda官网或Miniconda官网下载适合您系统的安装包,然后按照安装向导进行安装。
2. 创建一个新的conda环境
打开终端,输入以下命令创建一个名为“pytorch”的新环境:
```
conda create --name pytorch
```
3. 在新环境中安装PyTorch
激活新环境:
```
conda activate pytorch
```
然后,根据您的CUDA版本和Python版本选择适当的PyTorch版本,并使用以下命令安装:
```
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
```
如果您的系统没有安装CUDA,则可以使用以下命令安装CPU版本的PyTorch:
```
conda install pytorch-cpu torchvision-cpu -c pytorch
```
安装完成后,您可以在Python中导入PyTorch并开始使用它:
```
import torch
```
希望这些步骤能够帮助您在Ubuntu 18.04上成功安装PyTorch。
阅读全文