ubuntu linux安装pytorch和torchvision

1.下载镜像(镜像网址:https://download.pytorch.org/whl/torch_stable.html

假设你要下载torch1.4.0版本,cp36代表你的环境是python3.6,cu100代表的是你的cuda是10.0版本

 2.查看torch1.4.0对应的torchvision版本

Previous PyTorch Versions | PyTorch

3.已知torch1.4.0对应torchvision0.5.0

4.安装打开项目文件的终端

分别输入(要放在你的项目文件中,终端前面目录也要对)

pip install torch-1.4.0+cu100-cp36-cp36m-linux_x86_64.whl

pip install torchvision-0.5.0+cu100-cp36-cp36m-linux_x86_64.whl

5.完成

(最后安装完了才知道,虚拟机不能用cuda。。)

在WSL(Windows Subsystem for Linux)环境中安装CPU版本的PyTorch torchvision,你可以按照以下步骤操作: 1. **更新系统**: 首先,确保你的Ubuntu发行版已更新到最新状态,可以运行`sudo apt-get update && sudo apt-get upgrade`. 2. **安装依赖库**: 由于PyTorch需要一些额外的库支持,如gcc、cmake等,可以运行以下命令安装它们: ``` sudo apt-get install -y build-essential cmake libopenblas-dev luarocks libncurses5-dev libncursesw5-dev libreadline6-dev libdb5.3-dev libgdbm-dev liblzma-dev tk-dev libffi-dev python3-dev libssl-dev libsqlite3-dev libcurl4-openssl-dev libtbb-dev libjpeg-dev libpng-dev libwebp-dev libzmq3-dev ``` 3. **下载安装PyTorch**: 打开终端并导航到你想存放PyTorch的目录,然后从官方仓库下载CPU版本的PyTorch: ``` cd ~ wget https://download.pytorch.org/whl/cpu/torch-<version>-cpu.tar.gz tar xvf torch-<version>-cpu.tar.gz ``` 将 `<version>` 替换为你想要的PyTorch版本号。 4. **安装torchvision**: 安装完成后,进入刚解压的PyTorch文件夹,找到 `torch torchvision` 文件夹,同样下载对应版本的 torchvision whl 包并安装: ``` cd torch wget https://download.pytorch.org/whl/cpu/torchvision-<version>-cpu.tar.gz tar xvf torchvision-<version>-cpu.tar.gz ``` 然后将 `torchvision` 目录移动到 `python3.9/site-packages` 下,替换原来的文件: ``` mv torchvision-<version>/* /usr/local/lib/python3.9/dist-packages/ ``` 5. **验证安装**: 最后,检查安装是否成功,可以输入以下命令测试PyTorchtorchvision: ```bash python3 -c "import torch; print(torch.__version__)" python3 -c "from torchvision import transforms; print(transforms.__version__)" ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值