search

MiniBullLab / easy_docker

1 stars 1 forks source link

在T4上跑docker镜像训练报错 #18

Closed foww-0001 closed 4 years ago

foww-0001 commented 4 years ago

由于T4的torch需要用源码安装，而镜像中是用打包好的库安装，因此报错。

foww-0001 commented 4 years ago

运行以下命令解决错误：

sudo pip3 uninstall torch
sudo pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl

后续在镜像中需要考虑不同显卡的支持范围。

修正安装命令:

sudo pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp36-cp36m-linux_x86_64.whl

@lpj0822 @vitahlin