@conda安装torch+torch_geometric
一、 基础命令
- 查看当前存在哪些虚拟环境
conda env list #查看当前存在哪些虚拟环境 或 conda info --envs
带*的表示当前环境
- 创建虚拟环境
conda create -n [gnnEnv] python=3.9 #创建虚拟环境
[ ] 中为虚拟环境的名称
conda activate [gnnEnv] #激活需要进入的虚拟环境(Linux要用source activate)
- 查看已经安装的包
conda list # 查看已经安装的包
二、设置国内镜像源
- Conda 安装好以后,默认的源(即包下载地址)往往访问不稳定,下载速率慢,换到国内的镜像源下载速度就可以提高很多。下面是清华源的配置办法:
- 重新打开 Miniconda3 命令提示符窗口,输入 conda config --set show_channel_urls yes 命令:
- 回车后就会在用户名目录下创建了一个名为 .condarc 的文件,找到这个文件后以记事本方式打开,直接在文件中修改
- 或者输入命令 conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
三、 安装pytorch
- 配置清华源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes#下载时显示文件来源
- 配置完成后,进入pytorch官网
可以用nvcc --version
查看安装的CUDA编译器版本。
按照自己的配置进行选择
复制下载命令如下:
conda install pytorch torchvision cudatoolkit=11.3 -c pytorch
安装时候一定不要直接用这条命令,除非你vpn了,否则总是下载失败。问题就在这条命令上,最后的-c pytorch参数的意思是强制在pytorch官网下载。就是因为这一参数倒置了下载失败。
执行如下命令,即可以安装成功:
conda install pytorch torchvision cudatoolkit=11.3
在pytorch的世界遨游吧!在家折腾了一天折腾出来的经验!短短几句话,都是泪啊!
四、torch_geometric安装
- 确认自己的torch和cuda版本
import torch
print(torch.__version__) #检查tuorch版本
print(torch.version.cuda) #检查cuda版本
-
Pytorch官方链接:https://pytorch-geometric.com/whl/
根据上面查询的版本进行对应链接
-
下载whl文件
Python的版本是3.9.0,所以选择对应的whl文件分别点击进行下载。(注意:cp37m对应3.9.0,win对应Windows,amd64对应64位)
下载好文件后将这些whl文件复制粘贴到Anaconda中的Scripts文件夹中,方便后面使用pip进行安装。、 -
依次执行下面的命令
pip install torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl
pip install torch_sparse-0.6.12-cp37-cp37m-win_amd64.whl
pip install torch_cluster-1.5.9-cp37-cp37m-win_amd64.whl
pip install torch_spline_conv-1.2.1-cp37-cp37m-win_amd64.whl
pip install torch-geometric
- 验证是否安装成功
五、Anaconda删除虚拟环境
方法一:
-
第一步:首先退出环境
conda deactivate -
第二步:查看虚拟环境列表,此时出现列表的同时还会显示其所在路径
conda env list -
第三步:删除环境
conda env remove -p 要删除的虚拟环境路径
conda env remove -p /home/kuucoss/anaconda3/envs/tfpy36 #我的例子
方法二:
-
第一步:首先退出环境
conda deactivate -
第二步:删除环境
conda remove -n 需要删除的环境名 --all
六、其它问题
- 我们在使用 pip 安装 python 包时,经常会出现如下错误:
ERROR: Could not find a version that satisfies the requirement xxxx(from versions: none)
ERROR: No matching distribution found for xxxx
直接选用pip源并且信任它的来源就可以解决这种问题。
pip install 库包名 -i [http://pypi.douban.com/simple/](http://pypi.douban.com/simple/) --trusted-host [pypi.douban.com](http://pypi.douban.com/)
上面使用了豆瓣源,将其换成清华源、阿里源等都适用。
- 镜像源安装其它库
scikit-learn
pip install sklearn -i https://pypi.tuna.tsinghua.edu.cn/simple
pandas
pip install pandas -i https://mirrors.aliyun.com/pypi/simple/ --trusted-host=mirrors.aliyun.com/pypi/simple
rdkit
conda install -c conda-forge rdkit
- 查看自动电脑是否安装cuda
nvidia-smi