分cpu和GPU版本,因为我的机子有独显,所以我这里装的是GPU版本
1.下载cuda
在命令指示行窗口输入
nvidia-smi
我的机子cuda版本为12.3,算是比较高的那个,但是pytorch官网现阶段只放出了cuda11.8和12.1,建议11.8,因为有的不向下兼容
然后打开这个命令提示行
键入conda create --name pytorch_env python=3.11
后面有几个提示问你创建虚拟环境需要这些包,你输入y并且回
分cpu和GPU版本,因为我的机子有独显,所以我这里装的是GPU版本
1.下载cuda
在命令指示行窗口输入
nvidia-smi
我的机子cuda版本为12.3,算是比较高的那个,但是pytorch官网现阶段只放出了cuda11.8和12.1,建议11.8,因为有的不向下兼容
然后打开这个命令提示行
键入conda create --name pytorch_env python=3.11
后面有几个提示问你创建虚拟环境需要这些包,你输入y并且回