服务器pytorch相关配置(记录自用)

使用的是pycharm远程控制服务器,服务器内已安装并配置好miniconda3+python3.8.5。
新建环境后总是要配置一些东西,记录在此便于以后参考。

使用conda虚拟环境

#获取版本号 conda --version 或 conda -V
#检查更新当前conda conda update conda
#查看当前存在哪些虚拟环境 conda env list 或 conda info -e

conda create -n your_env_name python=X.X
source activate your_env_name //启动虚拟环境
source deactivate //关闭conda环境
conda remove -n your_env_name(虚拟环境名称) --all //删除conda虚拟环境
新建环境

新环境安装必要的包

0. 更新pip

python -m pip install --upgrade pip

使用命令 pip list 即可查看已安装的库列表:

1. pytorch

安装方法1

校园网下,直接根据官网指引,选择所对应的命令即可,无需更换源。https://pytorch.org/
我用的pip指令,而不是conda。
官网安装指引
检验是否成功

python
import torch
print(torch.__version__)  #查看版本号

终端界面显示如下
在这里插入图片描述

安装方法2(推荐)

后来发现通过 方法1 官网指引安装pytorch,可用版本比较少,比如我是CUDA11.0,但官网只有关于CUDA11.1的推荐。实际上这两个适配的pytorch版本是有一些差异的,在某些场景下可能会报错。因此我卸载了原本安装的pytorch,采用了另一种更灵活的方法。

  1. https://download.pytorch.org/whl/torch_stable.html 直接在该网页找到合适版本的torch并下载,然后拖到服务器内自己的home文件夹。
    根据命名就可以找到合适的版本,如下图所示,torch-1.7.1是pytorch的版本号,cu110指CUDA 11.0,cp38指python 3.8,最后是linux 64位系统。
    在这里插入图片描述

  2. 把下载下来的文件拖入服务器home文件夹内后,直接用命令安装即可
    pip3 install “刚才下载的文件名”(如 pip3 install torch-1.7.1-cp38-cp38-win_amd64.whl

  3. 查看pytorch和cuda是否匹配

python
import torch
torch.cuda.is_available()

如果返回True,则匹配;返回False,需要调整一下cuda或torch的版本,重新部署一下环境。

2. python安装numpy与scipy

安装了python3.0以上版本会比较简单
直接

pip install numpy
pip install scipy

3. 安装机器学习python库sklearn

首先需要进行包依赖检查,具体有以下几个要求:

  • Python(>=2.6 or >=3.3)
  • NumPy(>=1.6.1)
  • SciPy(>=0.9)

满足上述条件后,可以直接使用pip进行安装:

pip install -U scikit-learn
对于PyTorch服务器配置,以下是一些建议: 1. 操作系统:选择一个适合你的任务的操作系统,如Ubuntu、CentOS等。通常,Ubuntu是一个流行的选择,因为它有广泛的软件支持和社区支持。 2. 硬件要求:PyTorch可以在CPU和GPU上运行,但在进行深度学习训练时,使用GPU可以显著提高性能。因此,建议选择一台配备强大GPU的服务器。NVIDIA的GPU是常见的选择,像Tesla V100、RTX 2080 Ti等,但具体取决于你的预算和任务需求。 3. CUDA和cuDNN:PyTorch依赖于CUDA和cuDNN来加速GPU计算。确保在服务器上安装了与你所选GPU兼容的适当版本的CUDA和cuDNN。可以在NVIDIA官网上找到相应的版本和安装指南。 4. Python环境:安装适当的Python版本,并使用管理工具(如pip或conda)安装PyTorch和其他必要的依赖项。可以参考PyTorch官方文档获取安装指南。 5. 存储:确保服务器上有足够的存储空间来存储数据集、模型和其他必要的文件。使用SSD硬盘可以提高数据读取速度。 6. 内存:为了处理大型数据集和复杂模型,服务器需要足够的内存。内存的大小取决于你的任务需求。 7. 网络连接:如果需要从互联网上下载数据集或与其他服务器进行通信,确保服务器有稳定的网络连接。 8. 安全性:考虑服务器的安全性,如使用防火墙、设置密码、限制远程访问等。 这些只是一些常见的建议,具体的配置需求可能因任务的复杂性和预算而有所不同。根据你的需求和资源情况,可以进一步优化服务器配置
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值