1. Anaconda创建PyTorch环境
在搜索栏输入“Anaconda prompt”,点击进入对应窗口,执行如下操作:
① 修改镜像源提升下载速度:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
② 查看Python版本号:输入“python --version”返回版本号;
③ 创建Pytorch环境:
1)默认位置创建:直接输入conda create -n PyTorch python=xxx,PyTorch为环境名称,xxx为Python版本号,enter后按“y”→输入conda info --envs查看所有环境→输入conda activate PyTorch激活环境→打开Pytorch官网https://pytorch.org/→滑动到如下界面,选择conda,复制安装命令(或点击下方的Previous versons of PtTorch 选择其他版本)→在Anaconda prompt窗口输入刚才复制的命令,选择“y”下载资源。
2) 其他盘创建:a.在某盘创建PyTorch文件夹,如D:\Anaconda3\envs\PyTorch,b.返回Anaconda prompt界面,输入:conda create --prefix=创建的位置 python=xxx,如:conda create --prefix=D:\Anaconda3\envs\PyTorch python=3.11,enter后按“y”→输入conda info --envs查看所有环境→输入conda activate 文件位置,如D:\Anaconda3\envs\pytorch激活环境→打开Pytorch官网https://pytorch.org/→滑动到如下界面,选择conda,复制安装命令(或点击下方的Previous versons of PtTorch 选择其他版本)→在Anaconda prompt窗口输入刚才复制的命令,选择“y”下载资源。
④ 测试PyTorch:在Anaconda prompt界面输入
import torch
torch.cuda.is_available()
返回True说明为GPU版本,返回False为CPU版本。
2. Spyder添加PyTorch环境
① 打开Spyder做如下操作:
②在控制台输入import torch并回车,按提示的错误进一步进行相关操作,如推荐在Anaconda prompt界面输入某些指令、需要改变该环境下的某些库的版本(在Anaconda Navigator的环境界面下操作)等。
③测试:输入Import torch,不报错则成功,否则失败。
3. JupyterLab添加PyTorch环境
打开Anaconda prompt界面,执行如下操作:
①输入conda info --envs查看所有环境→输入conda activate myenv,激活名为myenv的环境
②安装ipykernel环境,输入conda install ipykernel
③将需要的虚拟环境添加到Jupyter Lab的内核中,输入:python -m ipykernel install --user --name=myenv --display-name="Python myenv",其中,myenv是刚才激活的环境
④打开JupyterLab,单击左上角菜单栏的Kernel→Change Kernel...→下拉选择需要的环境→Select,界面右上方则显示环境名
⑤测试:输入Import torch,不报错则成功,否则失败。