ubuntu18.04安装pytorch1.7.1
时间: 2023-04-26 18:05:02 浏览: 125
Ubuntu 18.04上安装 phpMyAdmin的详细教程
1. 首先,打开终端并更新软件包列表:
sudo apt update
2. 安装pip:
sudo apt install python3-pip
3. 安装pytorch:
pip3 install torch==1.7.1+cpu torchvision==.8.2+cpu -f https://download.pytorch.org/whl/cu101/torch_stable.html
4. 安装完毕后,可以在Python中导入pytorch并开始使用:
import torch
如果需要使用GPU版本的pytorch,请将上述命令中的“cpu”替换为“cu101”或“cu102”,具体取决于您的CUDA版本。
阅读全文