配置环境 本地+Linux服务器

安装anaconda

下载anaconda依赖文件(类似应用程序的exe?)从Index of / (anaconda.com)

进入该文件所在目录:sh ······.sh

服务器非root安装anaconda参考:

linux下非root用户安装Anaconda - 丁11 - 博客园 (cnblogs.com)

(3条消息) linux服务器成功卸载和安装anaconda(详细)_linux卸载anaconda3并重装_嗨,紫玉灵神熊的博客-CSDN博客

创建新的环境

conda create -n new_env_name python=3.7(python版本自己指定)

删除:conda env remove -n new_env_name

激活新的环境

conda activate new_env_name

【(new_env_name)[dir] 】

安装指定版本 pytorch、torchvision、cuda:!需要版本匹配!

1、

cpu版本下载指定版本的wheel文件从:https://download.pytorch.org/whl/torch;https://download.pytorch.org/whl/torchvision

cuda版本:

pip3 install --user torch==1.3.0+cu100 torchvision==0.4.1+cu100 -f https://download.pytorch.org/whl/cu101/torch_stable.html

(其中cu100代表cuda10.0.0,以此类推)

安装torch_gpu时依赖文件太大,可以:

①换源:pip --default-timeout=1000  install -U nvidia-cublas-cu11==11.10.3.66 -i https://pypi.tuna.tsinghua.edu.cn/simple

②百度whl文件名、到镜像网站下载

 百度获得→

https://pypi.tuna.tsinghua.edu.cn/packages/dc/30/66d4347d6e864334da5bb1c7571305e501dcb11b9155971421bb7bb5315f/nvidia_cudnn_cu11-8.5.0.96-2-py3-none-manylinux1_x86_64.whl

2、cd wheel文件所在路径

3、pip install ·······.whl

4、检查torch

        命令:python(进入py临时文件)

        >>import torch

        >>print(torch.__version__)

注意:必须先装torch再装torchvision,因为torchvision依赖torch

注意:期间遇到下载某些依赖wheel时报错(服务器上,因为网络不稳定),去官网下载对应报错的库的wheel,或pip install -i https://pypi.tuna.tsinghua.edu.cn/simple 库名

https://download.pytorch.org/whl/torch;(Pillow wheel: Pillow · PyPI)

查看cudnn版本:

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

Python3: print(torch.backends.cudnn.version())

查看cuda

nvidia-smi

Python3: print(torch.version.cuda)

torch/torchvision/python对应版本查询:torchvision · PyPI

查看cuda安装版本与路径

ls -l /usr/local | grep cuda

测试显卡与cuda是否适配

from torch import Tensor

    import torch

    x = torch.randn(2, 4).cuda()

    y = torch.randn(4, 1).cuda()

    out = (x @ y)

    print(out.shape)

print('Success, no Cuda errors means it worked see:\n')

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值