首先需要根据自己电脑的cuda版本安装查找对应的pytorch,下面的表格就是cuda版本对应的pytorch版本。
由于我的cuda版本是11.6,安装pytorch1.9.1即可。
首先需要创建新环境
其中monai是我的环境的名称,安装的python版本是3.6.5,然后按yes就可以创建新环境。
环境创建好后,下一步激活环境。再安装pytorch
由于我的cuda版本是11.6,安装torch版本是1.9.1即可。
首先需要根据自己电脑的cuda版本安装查找对应的pytorch,下面的表格就是cuda版本对应的pytorch版本。
由于我的cuda版本是11.6,安装pytorch1.9.1即可。
首先需要创建新环境
其中monai是我的环境的名称,安装的python版本是3.6.5,然后按yes就可以创建新环境。
环境创建好后,下一步激活环境。再安装pytorch
由于我的cuda版本是11.6,安装torch版本是1.9.1即可。