包的安装pytoch以及其他

pytorch利用pip安装教程

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html

查看服务器的cuda版本

ls -l /usr/local | grep cuda

之后在pytorch官网找到torch和torchvision对应的版本，在https://download.pytorch.org/whl/torch_stable.html该网址找到torch和你的电脑cuda对应的版本。

# 如果安装的cuda版本不匹配
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

pip查看可安装的包的版本

以scikit-learn为例

pip install scikit-learn==

输入命令后显示的结果：

note: This error originates from a subprocess, and is likely not a problem with pip.

我在安装scikit-learn==0.23.2时报错，采取的解决办法如下：

1. 给python降级，我原型安装的版本是3.10.2，现在换成3.7版本。

conda install python==3.7

成功安装包