安装PyTorch 1.9.0 Python 3.8(ubuntu18.04) Cuda 11.1
时间: 2025-01-02 12:44:47 浏览: 68
安装PyTorch 1.9.0版本以及CUDA 11.1在Ubuntu 18.04上需要遵循几个步骤:
1. **更新系统**:
首先确保你的系统是最新的,可以运行以下命令:
```bash
sudo apt-get update
sudo apt-get upgrade
```
2. **安装必要的依赖**:
对于Python 3.8,你需要安装`python3.8-dev`和`build-essential`等工具包,用于构建库:
```bash
sudo apt-get install python3.8-dev cmake libopenblas-dev luarocks libncurses5-dev libncursesw5-dev libreadline6-dev libdb5.3-dev libgdbm-dev libsqlite3-dev tk-dev libffi-dev
```
3. **安装cuDNN**:
CUDA 11.1通常会包含cuDNN,你可以检查已有的版本:
```bash
dpkg -l | grep cudnn
```
如果cuDNN已存在且版本合适,跳过这一步;如果需要更新,可以从NVIDIA官网下载cuDNN并安装。
4. **安装CUDA**:
下载CUDA Toolkit 11.1,按照官方文档进行安装,注意选择对应Ubuntu 18.04的版本。
5. **配置环境变量**:
打开`~/.bashrc` 或 `~/.bash_profile` (取决于你的Shell),添加CUDA路径到`PATH`环境变量,并设置CUDA_HOME指向安装目录:
```bash
export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-11.1
source ~/.bashrc
```
6. **安装PyTorch**:
使用conda创建一个新的环境,然后安装PyTorch 1.9.0和cuDNN支持:
```bash
conda create -n pytorch_1_9 python=3.8 torchvision cudatoolkit=11.1 -c pytorch
conda activate pytorch_1_9
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/torch_stable.html
```
7. **验证安装**:
完成上述步骤后,你可以通过以下命令测试安装是否成功:
```bash
python -c "import torch; print(torch.cuda.is_available())"
```
阅读全文