文章目录
默认CUDA已经安装,并查看其版本号
sudo nvidia-smi
一、Anaconda的下载安装
Anaconda官网主页:https://anaconda.org/
https://www.continuum.io/downloads
到官网进行下载,之后使用下面命令安装
bash Anaconda3-xxxx.sh
一直按回车,知道出现yes位置,之后会默认安装。
其中xxx就是对应的版本号
二、查看自己的Python版本
python
得到自己python版本号
三、安装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/
转载文章受原作者版权保护。转载请注明原作者出处!