pytorch 0.4.1+ubuntu18.04 +python3.67+cuda9.2 安装

原创

于 2018-10-29 22:50:24 发布 · 5.5k 阅读

CC 4.0 BY-SA版权

文章标签：

本文介绍了如何在Ubuntu 18.04操作系统上安装PyTorch 0.4.1版本，特别是针对配备CUDA 9.2的GPU进行安装。提供了通过pip和Conda两种方法进行安装的详细步骤，并且包含了安装后的测试验证方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

pytorch 0.4.1+ubuntu18.04 +python3.67+cuda9.2 安装

点击进入官网 https://pytorch.org/
方法一：pip安装

Python3.6+pip安装cpu版本

pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision

Python3.6+pip安装gpu版本
目前gpu版本支持cuda8.0,cuda9.0和cuda9.2，请选择对应的版本下载安装，不要同时执行下面三个命令！

pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision

（我选的是这个）
方法二：Conda安装

如果你是Anaconda|Python用户，cpu版本执行命令:

conda install pytorch-cpu torchvision-cpu -c pytorch

就可以完成安装。

,如果你需要安装cuda8.0、cuda9.0或cuda9.2的gpu版本，请执行：

conda install pytorch torchvision cuda92 -c pytorch

pip3 install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl

测试安装是否成功
python命令下输入：

import torch

print(t

200万优质内容无限畅学