当前位置:   article > 正文

ubuntu配置 pytouch_ubantu怎么打开pytouch

ubantu怎么打开pytouch

1. CPU 环境

pytouch 官网

1.1 进入官网
选择对应的基本信息,下方即可生成相应的安装命令;
1.2 键入对应命令

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
  • 1

2. GPU 环境

网址:CUDA Toolkit 10.1 update2 Archive

wget https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run
  • 1
  • 2
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/55559
推荐阅读
相关标签
  

闽ICP备14008679号