Avatar
滔滔设计侠
安装pytorch指南
1.搜索prompt打开
2.输入conda create -n pytorch python=3.9.12
其中这是创建的格式conda create -n
"pytorch"是这个环境变量的名字
"python=3.9.12"是我们要确定的当前python安装的版本数
输入y确认安装
3.输入activate pytorch
将base基础环境换成pytorch
输入pip list,查看当前环境下面有哪些包,我们发现没有pytorch,那么下面我们就需要安装它。
4. 输入nvidia-smi
查看CUDA version
去pytorch官网中下载的版本号应小于CUDA version号
如我的CUDA version为11.6
所以后续下载pytorch 11.3是合理的
5.https://pytorch.org/
打开网址是pytorch官网,点击install
选择合适的版本号,复制方框里的内容
6.粘贴至pytorch环境中
7.输入y确认下载安装
8.done显示安装成功
在输入pip list 检查torch
9.import torch 检查cuda
使用情况为true就对啦
恭喜pytorch安装完成✅
发布于 2024-12-22