创建新的conda虚拟gpu环境
时间: 2024-09-29 14:11:24 浏览: 44
Anaconda配置pytorch-gpu虚拟环境的图文教程
创建一个新的Conda虚拟环境并且指定使用GPU,通常需要以下几个步骤。假设你已经安装了`miniconda`或`conda-forge`:
1. **安装 Anaconda** (如果你还没有安装):
```bash
# 如果你是Windows用户
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Windows-x86_64.sh -O miniconda.sh
# 如果你是Linux/Mac用户
curl -L https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -o miniconda.sh
```
2. **运行安装脚本并创建新环境**:
```bash
bash miniconda.sh
# 接受默认配置,然后输入新环境的名称,例如"my_gpu_env"
conda create -n my_gpu_env python=3.7 # 或者选择你需要的Python版本
```
3. **激活新环境**:
对于Windows用户:
```bash
conda activate my_gpu_env
```
对于Linux或Mac用户:
```bash
source activate my_gpu_env
```
4. **安装CUDA和PyTorch for GPU**:
```
conda install cudatoolkit=10.2 # 根据实际可用的cuda版本替换
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
```
注意检查CUDA版本与PyTorch是否兼容。
5. **验证安装**:
```bash
python -c "import torch; print(torch.cuda.is_available())"
```
阅读全文