Ubuntu在Anaconda中创建虚拟环境并安装pytorch

最近安装了双系统,并在ubuntu系统中安装了各种环境,现在记录一下安装pytorch的过程和遇到的一些问题

1 首先,我们去清华镜像网站下载Anaconda,因为直接从官网下载速度很慢很慢。网站下载

我下载的版本是  Anaconda3-2022.05-Linux-x86_64.sh

下载好的anaconda如图所示(安装包选择默认的下载位置就好了,具体anaconda下载时候会让你选择它的安装路径)

之后进入终端,输入安装命令

bash Anaconda3-2022.05-Linux-x86_64.sh

之后输入yes,然后不断按回车,直到下一个Y/N,选择yes

 之后就是选择anaconda的安装路径

Anaconda3 will now be installed into this location:

[/home/aaa/anaconda3] >>>

#按回车的话就会安装到这个默认路径,如果要修改路径在后面输入自定义的路径

[/home/aaa/anaconda3] >>> /home/aaa/developer_tools/anaconda3 #自定义路径

 之后就是继续输入yes然就安装完成了。

2 创建虚拟环境

创建名为pytorch的虚拟环境来安装Pytorch,python版本为3.7

conda create -n pytorch python=3.7 #创建名为pytorch的anaconda虚拟环境

(建议安装3.7版本的,之前安装了3.6版本的之后下载pytorch默认下载的是cpu版本的,用torch.cuda.is_available()测试总是返回False找了很多方法都没用,后来重装了虚拟环境把python设置为3.7后就成功了)

 如果创建太慢甚至导致失败,这时候需要更换国内的镜像源(其实一般都要更换镜像,除了这的python3.7的下载,后面的Pytorch下载没有配置镜像几乎下载不下来

更换清华镜像源

方法一:

输入

sudo gedit ~/.condarc 

然后在里面添加清华镜像,点击右上角保存

  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

如图所示

方法二:在终端中输入如下命令直接添加

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

conda config --set show_channel_urls yes   # 显示包的下载来源




查看是否修改成功

conda config --show channels #查看是否修改成功

如下图所示代表镜像配置成功,就可以安装 

 

(如果后期想删除镜像的话)

#如果想删除的话
conda config --remove channels http://mirrors.aliyun.com/anaconda/pkgs/main/ 

 (PS:现在有的情况下创建环境报错:ProxyError: Conda cannot proceed due to an error in your proxy configuration. Check for typos and other configuration errors in any '.netrc' file in your home directory, any environment variables ending in '_PROXY', and any other system-wide proxy configuration settings.的话,

就把清华镜像换成中科大镜像,

http://mirrors.ustc.edu.cn/anaconda/pkgs/free/

或者有同学开了VPN就把VPN关了) 

 然后再创建虚拟环境就会很快

conda create -n pytorch python=3.7#创建名为pytorch的anaconda虚拟环境

 激活虚拟环境

创建完虚拟环境后,别忘了激活虚拟环境,不然Pytorch是安装在base环境里的

conda activate pytorch #激活名为pytorch的虚拟环境

3 安装Pytorch(在刚创建的名为pytorch虚拟环境里安装) 

安装对应自己CUDA版本的Pytoch,Pytorch官网

复制该界面最下方的安装命令(注意命令最后的 -c python不要复制进去,复制进去的话下载依然用的国外的网站,速度很慢很慢) 

 

 像下面一样复制就可以了

conda install pytorch torchvision torchaudio cudatoolkit=11.3

安装完成验证是否安装pytoch成功

#输入python

python

>>>import torch    #导入pytorch

>>>torch.__version__ #查看版本

>>>torch.cuda.is_available() #查看能否使用GPU

如果跟下图一样代表安装成功了

  • 11
    点赞
  • 54
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值