安装PyTorch

  1. 创建虚拟环境
    打开Anaconda Prompt,输入以下命令创建新环境:

    conda create -n pytorch_env python=3.8
    

    激活环境:

    conda activate pytorch_env
    

  2. 安装PyTorch
    访问PyTorch官网获取适用于当前系统的安装命令。例如,使用Conda安装CPU版本:

    conda install pytorch torchvision torchaudio cpuonly -c pytorch
    

    若需GPU版本(需提前配置CUDA),替换为:

    conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
    

  3. 验证安装
    启动Python解释器,运行以下代码:

    import torch
    print(torch.__version__)
    print(torch.cuda.is_available())  # GPU版本检查
    


配置Jupyter Notebook

  1. 安装Jupyter
    在已激活的虚拟环境中执行:

    conda install jupyter
    

  2. 添加内核到Jupyter
    确保当前处于目标虚拟环境,安装ipykernel并注册内核:

    pip install ipykernel
    python -m ipykernel install --user --name=pytorch_env
    

  3. 启动Jupyter Notebook
    输入以下命令启动:

    jupyter notebook
    

    在新建笔记本时,选择pytorch_env作为内核。


常见问题解决

  1. 内核无法连接
    检查内核是否注册:

    jupyter kernelspec list
    

    若缺失,重新执行ipykernel安装步骤。

  2. 依赖冲突
    使用以下命令清理冲突包:

    conda clean --all
    conda update --all
    

  3. GPU不可用
    确认CUDA版本与PyTorch版本匹配,运行nvidia-smi检查驱动状态。

解释一下这两行是干嘛的(一看就懂)

  1. conda install ipykernel→ 给环境装一个 Jupyter 能识别的内核,假设环境名字是torch,接下来:
  2. python -m ipykernel install --user --name torch --display-name "PyTorch 环境"→ 把这个环境注册到 Jupyter 里,让 Jupyter 能看到它
Logo

Agent 垂直技术社区,欢迎活跃、内容共建。

更多推荐