jupyter使用pytorch

1、激活环境

以下所有命令都在Anaconda Prompt中操作。

bash 复制代码
conda activate 环境名称

我的环境名称是myenv

如果不知道自己的pytorch配在哪个环境,就用下面方法挨个试。

2、安装jupyter

1、安装

bash 复制代码
pip install jupyter

2、如果已经安装,检查jupyter是否已经安装在此环境中

bash 复制代码
conda list 想要检查的包的名称

3、安装ipykernel

bash 复制代码
pip install ipykernel -i https://pypi.tuna.tsinghua.edu.cn/simple

4、将torch环境加入jupyter内核

bash 复制代码
python -m ipykernel install --name 环境名称

5、修改juypter默认运行环境

1、先新建一个空白文件夹,文件夹名称自己起(我这里叫jupyterFiles)

2、查看jupyter配置文件地址

bash 复制代码
jupyter notebook --generate-config

根据文件生成路径找到generate-config文件,并打开

3、在已经打开的文件中,使用快捷键ctrl+f 搜索 The directory to use for notebooks。大概是984行上下,去除注释并添加你新建的文件夹地址。如下图

(这个图里的路径不是我的路径,我应该写成E:\DeepLearning\jupyterFiles)

6、运行jupyter

bash 复制代码
jupyter notebook

7、点击new 选择你的torch环境

测试

bash 复制代码
import torch
print(torch.cuda.is_available())
相关推荐
智恒百亿6 小时前
5090 八卡服务器踩坑实录:架构、多卡通信、PyTorch Distributed 一次说清楚
服务器·pytorch·架构
田里的水稻7 小时前
PyTorch-Dataset\DataLoader数据加载
人工智能·pytorch·python
智码看视界7 小时前
深度学习:环境搭建PyTorch vs TensorFlow 选型
pytorch·深度学习·tensorflow·gpu·环境搭建·cuda·框架选型
论文复现现场8 小时前
InfiniteTalk 需要多少显存?RTX 3090 24GB 单卡如何验证 480P、排查 OOM
pytorch·模型部署·ai视频生成·显存优化·infinitetalk·rtx3090·cuda oom
Zip-List9 小时前
工作小记-pytorch的算子优化: 转gpu核函数 python调c
c语言·c++·pytorch·python·cuda
Smoothcloud润云1 天前
PyTorch推理输入长度分桶与Padding优化实战:减少无效计算
人工智能·pytorch·python·深度学习·gpu算力
开开心心就好1 天前
手机虚拟化学实验室,混合燃烧都能模拟
jupyter·智能手机·ffmpeg·计算机外设·电脑·音视频·visual studio
盼小辉丶1 天前
PyTorch强化学习实战(29)——深入解析强化学习中的探索策略
pytorch·深度学习·强化学习
FBI HackerHarry浩2 天前
Pytorch中 CPU tensor vs GPU tensor
人工智能·pytorch·python·深度学习
Smoothcloud润云3 天前
PyTorch推理服务模型热切换与失败回滚实战:降低版本更新中断
服务器·人工智能·pytorch·python·深度学习·gpu