Linux 安装PyTorch
时间: 2024-08-04 13:00:43 浏览: 61
在Linux上安装PyTorch通常需要几个步骤。以下是简要的指南:
1. 更新系统:
```sh
sudo apt-get update
sudo apt-get upgrade
```
2. 安装必要的依赖:
对于CUDA支持,你需要安装对应的库和工具包,例如CUDA、cuDNN和gcc。如果你不确定版本,请先确认你的CUDA版本,然后从官方GitHub仓库下载对应版本的PyTorch。
```sh
sudo apt-get install nvidia-cuda nvidia-cudnn libncurses5-dev build-essential cmake python3-dev python3-pip swig
```
3. 下载并配置PyTorch源码:
```sh
git clone https://github.com/pytorch/pytorch.git
cd pytorch
```
4. 配置安装选项,根据需要选择CPU或GPU安装,并指定Python路径(例如,对于CPU安装):
```sh
export TORCH_CUDA_ARCH_LIST="6.0"
./configure --prefix=$HOME/.local/packages/torch --cmake-force-advanced -DCMAKE_BUILD_TYPE=Release -DPYTHON_EXECUTABLE=/usr/bin/python3 -DUSE_NCCL=OFF
```
5. 编译并安装:
```sh
make -j$(nproc)
sudo make install
```
6. 验证安装:
```sh
python3 -c "import torch; print(torch.__version__)"
```
如果遇到问题,可能需要查看PyTorch的官方文档或社区资源,因为某些发行版的依赖可能有所不同。
相关推荐
![docx](https://img-home.csdnimg.cn/images/20210720083331.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)