ubuntu16.04在CUDA9.1以及CUDA10.0下安装pytorch

测试CUDA

查看系统CUDA版本,方法:

  • cat /usr/local/cuda/version.txt
  • nvcc -V

具体见ubuntu16.04查看CUDA和cuDNN版本与Ubuntu14.04 查看安装好的CUDA

安装PyTorch

https://pytorch.org/

https://pytorch.org/get-started/previous-versions/

ubuntu16.04在CUDA9.1以及CUDA10.0下安装pytorch_第1张图片

ubuntu16.04在CUDA9.1以及CUDA10.0下安装pytorch_第2张图片


补充:cuda10.0 安装pytorch

  • pytorch 1.0稳定版正式发布之前,使用下面的命令安装pytorch,版本是pytorch 0.4.1

conda install pytorch torchvision -c pytorch
  • 2018.12.7在 NeurIPS 大会上,Facebook 正式发布 PyTorch 1.0 稳定版。新的安装命令如下

conda install pytorch torchvision cuda100 -c pytorch

ubuntu16.04在CUDA9.1以及CUDA10.0下安装pytorch_第3张图片


 验证是否安装成功 

ubuntu16.04在CUDA9.1以及CUDA10.0下安装pytorch_第4张图片

你可能感兴趣的:(Linux,深度学习)