在Anaconda Jupyter notebook中配置pytorch
Step 1: 创建环境
进入anacoda之后默认是base环境
创建新环境
conda create -n <name> python=<python的版本>
查看当前所有环境
conda info --envs
进入刚才创建环境
conda activate <name>
退出当前环境
conda activate
删除所指环境
conda remove -n <name> --all
Step 2:在创建的环境里面安装pytorch
进入pytorch官网,选择电脑配置
windows下推荐conda,Linux下推荐pip
我的电脑是有GPU的,看版本的时候选择CUDA对应版本,需要确定自己电脑显卡的版本和算力,我的电脑算力为6.1,这个可以满足CUDA 9 10 11的版本
随便打开一个命令行窗口,nvidia-smi命令查看Cuda版本,我的电脑的CUDA Version为12.4
于是复制对应的代码
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
这个时候遇到一个pytorch安装失败的问题,原因是没有给文件夹管理员权限,打开所在位置文件夹,右键属性–安全–编辑,给自己当前用户赋予管理员权限
如果安装的太慢可以使用镜像
从命令行输入python打开python代码界面,尝试import torch看是否报错,不报错说明之前的安装是成功的
输入torch.cuda.is_available() 如果为True就意味着可以使用GPU
Step 3:将下载好软件的pytorch环境配置内核到jupyter notebook
以管理员身份打开Anaconda Prompt
在base环境中输入安装
conda install nb_conda_kernels
进入创建的环境,命令行输入
conda install ipykernel
打开Jupyter Notebook会发现增加了两项:Python [conda env:pytorch] 和 Python [conda env:root]*
点击Kernel-Change kernel-Python[conda env:pytorch]即可创建成功
3253

被折叠的 条评论
为什么被折叠?



