Linux下如何安装pytorch
时间: 2024-12-25 12:32:42 浏览: 101
在Linux系统上安装PyTorch,通常可以按照以下步骤操作:
1. **更新包列表**:
```sh
sudo apt-get update
```
2. **安装依赖项**:
PyTorch需要一些额外的库,如CUDA(如果你计划在GPU上运行)、cuDNN、Python以及其 development headers。例如,如果你有nVIDIA GPU,你需要运行:
```sh
sudo apt-get install python3-dev python3-pip libncurses5-dev \
libfreetype6-dev libglib2.0-dev libbz2-dev libreadline-dev \
libsqlite3-dev libssl-dev libffi-dev liblzma-dev libcurl4-openssl-dev \
cmake pkg-config libopenblas-dev
```
如果你还需要CUDA和cuDNN,首先下载并安装它们,然后添加CUDA的路径到环境变量。
3. **下载和构建PyTorch源码**:
- 首先访问PyTorch官网 (https://pytorch.org/get-started/locally/) 获取最新版本的安装指南。
- 下载源码 tarball 或者通过 `pip` 安装预编译的 Whl 包。
```sh
# 使用HTTPS链接下载源码
wget https://github.com/pytorch/pytorch/releases/download/v1.9.0/torch-1.9.0%2Bcpu.tar.gz
# 解压并进入目录
tar xvf torch-1.9.0+cpu.tar.gz
cd torch-1.9.0+cpu
# 指定安装路径,比如 /usr/local
export CMAKE_PREFIX_PATH=/usr/local
# 构建并安装
mkdir build
cd build
cmake ..
make -j$(nproc) # 使用所有CPU核心进行编译
sudo make install
```
4. **检查安装**:
```sh
python3 -c "import torch; print(torch.__version__)"
```
请注意,这个过程可能会因操作系统版本、硬件配置和其他因素而略有差异。如果遇到问题,建议查看官方文档或者社区论坛寻求帮助。
阅读全文
相关推荐

















