Anaconda中Pytorch的安装及JupyterNoteBook的环境设置
·
安装PyTorch
-
创建虚拟环境
打开Anaconda Prompt,输入以下命令创建新环境:conda create -n pytorch_env python=3.8激活环境:
conda activate pytorch_env -
安装PyTorch
访问PyTorch官网获取适用于当前系统的安装命令。例如,使用Conda安装CPU版本:conda install pytorch torchvision torchaudio cpuonly -c pytorch若需GPU版本(需提前配置CUDA),替换为:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch -
验证安装
启动Python解释器,运行以下代码:import torch print(torch.__version__) print(torch.cuda.is_available()) # GPU版本检查
配置Jupyter Notebook
-
安装Jupyter
在已激活的虚拟环境中执行:conda install jupyter -
添加内核到Jupyter
确保当前处于目标虚拟环境,安装ipykernel并注册内核:pip install ipykernel python -m ipykernel install --user --name=pytorch_env -
启动Jupyter Notebook
输入以下命令启动:jupyter notebook在新建笔记本时,选择
pytorch_env作为内核。
常见问题解决
-
内核无法连接
检查内核是否注册:jupyter kernelspec list若缺失,重新执行
ipykernel安装步骤。 -
依赖冲突
使用以下命令清理冲突包:conda clean --all conda update --all -
GPU不可用
确认CUDA版本与PyTorch版本匹配,运行nvidia-smi检查驱动状态。
解释一下这两行是干嘛的(一看就懂)
conda install ipykernel→ 给环境装一个 Jupyter 能识别的内核,假设环境名字是torch,接下来:python -m ipykernel install --user --name torch --display-name "PyTorch 环境"→ 把这个环境注册到 Jupyter 里,让 Jupyter 能看到它
更多推荐


所有评论(0)