Ubuntu 18.0
PyTorch官网 https://pytorch.org/get-started/locally/ ,选择相对应的版本和系统信息,然后在服务器上执行命令即可
执行命令,一路yes完成
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
Ubuntu 18.0
PyTorch官网 https://pytorch.org/get-started/locally/ ,选择相对应的版本和系统信息,然后在服务器上执行命令即可
执行命令,一路yes完成
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch