ubunutu18.04安装pytorch
时间: 2023-10-23 20:15:04 浏览: 168
要在Ubuntu 18.04上安装PyTorch,您可以按照以下步骤进行操作:
1. 打开终端,并使用以下命令安装Miniconda(如果您已经安装了Conda,可以跳过此步骤):
```
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
```
2. 创建一个新的Conda环境并激活它:
```
conda create -n pytorch_env python=3.8
conda activate pytorch_env
```
3. 根据您的CUDA版本选择合适的命令进行安装。如果您的CUDA版本是10.2,可以使用以下命令:
```
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
```
如果您的计算机没有安装CUDA或者您希望安装CPU版本的PyTorch,可以使用以下命令:
```
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
4. 安装完成后,您可以使用以下命令验证PyTorch是否成功安装:
```
python
>>> import torch
>>> torch.__version__
```
如果没有报错并且能够打印出PyTorch的版本号,那么说明安装成功了。
请注意,以上步骤中的命令是基于使用Conda进行安装的。您也可以使用pip进行安装,但是使用Conda可以更好地管理包的依赖关系。
阅读全文