创建一个pytorch环境(CPU版本)

本文详细指导如何在Windows系统上使用Anaconda创建Fasterrcnn环境,包括安装不同版本的PyTorch(CPU/CUDA)及处理CUDAToolkit版本,同时介绍了如何通过pip安装其他依赖包和管理环境配置。
摘要由CSDN通过智能技术生成

1 打开Anaconda Prompt创建并激活

conda create -n fasterrcnn python=3.6

conda activate fasterrcnn

2 安装对应版本Pytorch

1 打开官网,​​​​​Previous PyTorch Versions | PyTorch

找到对应版本,如安装1.2.0:

使用以下可能报错 ~invalid

# CUDA 9.2
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch

# CUDA 10.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

# CPU Only
conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch
使用这个可以创建
Linux and Windows
# CUDA 10.0
pip install torch==1.2.0 torchvision==0.4.0

# CUDA 9.2
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

参考博主的博客指令:

pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

搜索深度学习环境配置5——windows下的torch-cpu=1.2.0环境配置_requirement里的torch-cpu_Bubbliiiing的博客-CSDN博客

3 安装其他包

在文件夹里建requirements,或者其他路径,如上述链接Bubbliiiing博客所述,在python终端运行

pip install -r requirements.txt

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值