ubuntu使用anaconda创建虚拟环境,并安装pytorch1.3

上篇介绍了如何安装Ubuntu安装NVIDIA驱动、CUDA CUDNN 环境,连接如下
https://blog.csdn.net/sinat_42239797/article/details/101618638
今天介绍如何安装pytorch
最好的安装方式是创建一个虚拟环境来安装,因为创建虚拟环境安装不成功或者将来不用了卸载的话比较方便,可以快速卸载并且卸载干净,这样对其他软件没有影响
首先创建一个虚拟环境

conda create -n your_env_name python=X.X(2.7、3.6等)   #your_env_name为你的虚拟环境的名字
eg:conda create -n pytorch python=3.7  #方便我创建虚拟环境名字为安装文件名字

等虚拟环境创即按好后要选择激活虚拟环境

conda activate pytorch

然后安装pytorch,我是在官方网站根据自己的环境选择的,官网链接https://pytorch.org
ubuntu使用anaconda创建虚拟环境,并安装pytorch1.3_第1张图片
根据给的安装提示在终端输入安装

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

安装完成后进行测试是否安装成功

python
import torch 
import torchvision

检测你的GPU是否可用(cuda是否可用)

import torch
torch.cuda.is_available()
输出为True则表示成功

如果不出错就是安装成功了
如果不编辑了退出虚拟环境

 source deactivate

将来在进入就再次激活
如果需要删除虚拟环境

conda remove -n your_env_name(虚拟环境名称) --all

你可能感兴趣的:(软件安装)