Pytorch环境简易搭建教程
- 输入nvidia-smi查看支持的cuda的最高版本(右侧:我的是12.4)。如果这个命令不行是没安装英伟达驱动,自行搜索安装
- 输入nvcc -V查看当前cuda版本,比刚刚的低或相等都行。如果命令不成功是没安装cuda,自行搜索安装
- 确定了cuda版本去pytorch.org官网选择配置安装,
-
复制图中红框命令安装,更早版本在左下角蓝框那里
-
输入以下命令确认安装成功
.
附加内容:如果想使用jupyter notebook,但遇到多个py虚拟环境,只下载了一个jupyter,想多个conda共享,原来的本办法是每个conda环境都安装一个jupyter,合理吗?不合理!
==办法:==jupyter notebook只是一个前端展示页面,关键是后面运行的内核:kernel,只需要在想使用的conda环境中安装ipykernel即可在notebook中切换该内核运行,不用再每个都安装jupyter了。
安装命令:pip install ipykernel