ubuntu中pytorch
时间: 2025-05-15 19:25:24 浏览: 22
### 如何在Ubuntu上安装和配置PyTorch
#### 准备工作
为了确保顺利安装PyTorch,在开始之前需确认已正确设置好CUDA环境。对于特定版本的CuDNN文件,可利用如下命令完成安装:
```bash
sudo dpkg -i cudnn-local-repo-ubuntu2204-8.6.0.163_1.0-1_amd64.deb
```
此操作适用于Ubuntu 22.04 LTS系统并准备相应版本的CuDNN库[^3]。
#### 下载Anaconda Installer Script
采用`curl`工具下载Anaconda安装脚本至本地:
```bash
curl --output anaconda.sh https://repo.anaconda.com/archive/Anaconda3-2022.05-Linux-x86_64.sh
```
上述命令会获取指定版本的Anaconda Linux发行版,并保存为`anaconda.sh`文件[^4]。
#### 执行Anaconda安装过程
赋予下载好的shell脚本执行权限,并启动安装流程:
```bash
chmod +x anaconda.sh
./anaconda.sh
```
按照提示完成安装向导中的各项选项设定;通常建议接受默认提议以便顺利完成整个过程。
#### 配置环境变量
一旦Anaconda成功部署完毕,则应调整`.bashrc`或其他Shell初始化文件来加入新路径:
```bash
echo 'export PATH="/home/user/anaconda3/bin:$PATH"' >> ~/.bashrc
source ~/.bashrc
```
这里务必把`"/home/user/"`替换成实际存放Anaconda目录的位置[^5]。
#### 创建专用Conda虚拟环境与激活
创建一个新的Python环境用于隔离不同项目间的依赖关系冲突:
```bash
conda create --name pytorch_env python=3.9
conda activate pytorch_env
```
这一步骤有助于保持全局环境中其他项目的纯净度不受影响。
#### 安装兼容版本的PyTorch及相关组件
最后借助于`conda`或官方推荐的方式引入目标框架及其必要的支持模块:
```bash
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
```
这条语句假设读者已经具备适当规格的NVIDIA GPU硬件设施以及相匹配驱动程序的支持下进行GPU加速计算[^1]。
阅读全文
相关推荐

















