当前位置:   article > 正文

ubuntu18.04+cuda10.0+anaconda3安装pytorch记录

ubuntu18.04+cuda10.0+anaconda3安装pytorch记录
  1. 首先使用root权限创建环境:
    sudo -s
    conda create-n 你的虚拟环境名 python=3.7

  2. 查询与自己安装的cuda对应的版本,由于直接在线安装较慢,有以下两种较快的方法:
    强烈推荐方法二,亲测好用
    1)方法一:可以先将torch-1.3.0和torchvision-0.4.1的.whl文件下载好,下载地址:
    https://download.pytorch.org/whl/torch_stable.html
    激活刚刚创建的环境,比如这里我的是:
    source activate pytorch
    进入环境后使用pip命令离线安装torch和torchvision:
    a. 进入下载目录,并安装torch:
    pip3 install torch-1.0.0-cp37-cp37m-linux_x86_64.whl

    (pytorch) root@yy:~# cd '/home/yy/下载' 
    (pytorch) root@yy:~/下载# ls
    Anaconda3-2020.02-Linux-x86_64.sh  cudnn-10.0-linux-x64-v7.4.2.24.tgz
    cuda                               torch-1.0.0-cp37-cp37m-linux_x86_64.whl
    cuda_10.0.130_410.48_linux.run
    (pytorch) root@yy:~/下载# pip3 install torch-1.0.0-cp37-cp37m-linux_x86_64.whl
    WARNING: The directory '/home/yy/.cache/pip' or its parent directory is not owned or is not writable by the current user. The cache has been disabled. Check the permissions and owner of that directory. If executing pip with sudo, you may want sudo's -H flag.
    Processing 
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/正经夜光杯/article/detail/937055
推荐阅读
相关标签
  

闽ICP备14008679号