服务器 cuda9.0.x 安装 pytorch1.2 稳定版 :
执行 conda install pytorch torchvision cudatoolkit=9.0 -c pytorch 安装cuda9.0对应的pytorch,相应的,cuda版本号应该可以任意更改。
服务器 cuda9.0.x 安装 pytorch1.2 稳定版 :
执行 conda install pytorch torchvision cudatoolkit=9.0 -c pytorch 安装cuda9.0对应的pytorch,相应的,cuda版本号应该可以任意更改。