ubuntu pytorch配置conda新环境
时间: 2024-10-09 20:09:48 浏览: 55
在Ubuntu系统上安装PyTorch并配置一个新的Conda环境,通常分为以下几个步骤:
1. 安装Miniconda:首先访问Anaconda的官方网站下载适用于Linux的Miniconda,选择64位版本,然后添加安装脚本到PATH。
```bash
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh -b -p $HOME/miniconda
export PATH="$HOME/miniconda/bin:$PATH"
```
2. 创建新环境:打开终端,创建一个新的Conda环境,比如命名为`pytorch_env`,并激活它。
```bash
conda create -n pytorch_env python=3.7 # 或者其他Python版本
source activate pytorch_env
```
3. 安装PyTorch:在激活的新环境中安装PyTorch,由于PyTorch的官方推荐安装方式是通过pip,可以这样操作:
```bash
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
```
这里假设你的CUDA版本是10.2,如果不确定可以查看CUDA官网或者NVIDIA驱动管理软件。
4. 验证安装:最后检查一下是否安装成功,运行以下命令:
```bash
python -c "import torch; print(torch.__version__)"
```
阅读全文