最近安装了双系统,并在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/
?然后再创建虚拟环境就会很快
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
如果跟下图一样代表安装成功了

?
|