ubuntu18.04安装pytorch

文章目录

默认CUDA已经安装,并查看其版本号

sudo nvidia-smi

ubuntu18.04安装pytorch

一、Anaconda的下载安装

Anaconda官网主页:https://anaconda.org/
https://www.continuum.io/downloads

到官网进行下载,之后使用下面命令安装

bash Anaconda3-xxxx.sh

一直按回车,知道出现yes位置,之后会默认安装。

其中xxx就是对应的版本号

二、查看自己的Python版本

python

得到自己python版本号

ubuntu18.04安装pytorch

三、安装pytorch

1.查看对应版本的pythorch并安装

官网:https://pytorch.org/get-started/locally/

我的版本对应的版本是11.1

安装代码如下:

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

注意:
1)这步容易因网速原因失败
比如:

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https:
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.

HTTP errors are often intermittent, and a simple retry will get you on your way.

Collecting package metadata (current_repodata.json): failed

通过更换源,可以解决这个问题:

#1、备份本地配置
cp ~/.condarc ~/.condarc.bak

sudo gedit  ~/.condarc
#2、删除原来的内容,填入以下
channels:
  - defaults
show_channel_urls: true
channel_alias: https:
default_channels:
  - https:
  - https:
  - https:
  - https:
  - https:
custom_channels:
  conda-forge: https:
  msys2: https:
  bioconda: https:
  menpo: https:
  pytorch: https:
  simpleitk: https:

#3、清除缓存
conda clean -i

2)有的教程让独立一个新的conda环境,进行pytorch安装
但是就我的实践看,容易出现问题。其原因是独立一个新的conda环境时,是将python直接复制过去的,所以一些接口路径就变了,而导致安装失败。当然你可以一个个文件查找并将路径改正。

2.测试

代码如下(示例):

>>> import torch
>>> torch._version_
Traceback (most recent call last):
  File "", line 1, in <module>
AttributeError: module 'torch' has no attribute '_version_'
>>> torch.__version__
'1.11.0'
>>>

Original: https://blog.csdn.net/zxnzjccmily/article/details/124660241
Author: 猪猪侠的猪猪女孩
Title: ubuntu18.04安装pytorch

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/707681/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球