Pytorch环境简易搭建教程

Pytorch环境简易搭建教程

1. 输入nvidia-smi查看支持的cuda的最高版本（右侧：我的是12.4）。如果这个命令不行是没安装英伟达驱动，自行搜索安装

2. 输入nvcc -V查看当前cuda版本，比刚刚的低或相等都行。如果命令不成功是没安装cuda，自行搜索安装

3. 确定了cuda版本去pytorch.org官网选择配置安装，

4. 复制图中红框命令安装，更早版本在左下角蓝框那里

5. 输入以下命令确认安装成功

.

---

附加内容：如果想使用jupyter notebook，但遇到多个py虚拟环境，只下载了一个jupyter，想多个conda共享，原来的本办法是每个conda环境都安装一个jupyter，合理吗？不合理！
==办法：==jupyter notebook只是一个前端展示页面，关键是后面运行的内核：kernel，只需要在想使用的conda环境中安装ipykernel即可在notebook中切换该内核运行，不用再每个都安装jupyter了。
安装命令：pip install ipykernel