如何正确配置pytorch环境

1 各种镜像源的问题

因为pytorch原生的服务器在国外,而pytorch(cuda版本)大小为1G以上,使用国外源下载是不太可能的。因此衍生出了很多国内的下载源。pytorch主要包括三个包,pytorch、torchversion和torchaudio,另外在加一个cudatoolkit。

清华源:

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

南京大学源:

https://mirrors.nju.edu.cn/anaconda/cloud/pytorch/win-64/

源的概念就是一个下载地址,在这个地址下,有需要的pytorch安装包,在此地址下就可以快速的安装好pytorch的环境。

2 下载源的切换

b站很多博主都是直接把一些源直接添加到anaconda的资源库里面,这样会造成一些混乱。因为不同的源里的资源是不同的,有的是用来下载pytorch的,可能还有一些是用来下载其他的必要的安装包的。因此在进行下载时,手动配置下载地址即可,默认下载地址保持不变。

配置下载地址的方法也很简单:

conda install pytorch==1.11.0 torchversion==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

只需要在-c 后手动把下载源的链接加上面就行了。

3 下载指令

下载指令有conda和pip,这两个都是下载的方式,有的使用conda,而有时候使用pip。这两个命令相当于从不同的软件商店下载同一个软件的关系。

4 本地的下载

也可以把pytorch下载到本地进行安装,下载文件格式为.whl

conda install --use-local dir
pip install dir

4 一些指令

conda list #用来查看虚拟环境中的安装包,各种插件
conda install #安装需要的安装包
conda uninstall #卸载需要的安装包
conda config --show #下载通道查看
conda config --get  #下载通道查看
conda config --add channels 下载源链接  #添加通道
conda config --remove channels 链接     #移除通道
conda install nb_conda #安装jupyter
jupyter notebook #用来启动安装好的jupyter
conda/pip install opencv-python #opencv的安装

5 pytorch准确安装检验

python #进入python环境
import torch #进入torch
torch.cuda.is_available() #验证是否正常使用
true

你可能感兴趣的:(pytorch,人工智能,python)