anoconda配置pytorch

该文提供安装PyTorch的详细步骤,包括在Anaconda环境下创建环境、选择Python版本、安装不同CUDA版本的PyTorch以及CPU版本。同时,文章还介绍了如何解决编译时CUDA错误的问题,并提到了安装scikit-learn库的方法。
摘要由CSDN通过智能技术生成

安装pytorch

参考:https://blog.csdn.net/Guoyingyuan/article/details/124497041

Anaconda promot(在Anaconda promot窗口内执行)
(1)

python -m pip install -U pip
conda create --name pytorch3.7 python=3.7

Anaconda promot
在这里找需要版本的pytorch安装命令,带CUDA即为GPU版本
https://pytorch.org/get-started/previous-versions/
https://pytorch.org/get-started/locally/
(2)

activate pytorch3.7
# CUDA 11.3
(torch1.12.1)
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
(torch1.11.0)
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

Anaconda promot
(3)安装sklearn库

activate tensorflow
pip install scikit-learn

出现乱码的解决方法

Pycharm控制台输出乱码——已解决

(4)安装cpu版本

conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cpuonly -c pytorch

出现错误:assertionerror: torch not compiled with cuda enabled ( Solution )

解决方案1:从CUDA切换到CPU版本 –

通常在PyTorch中编译任何神经网络时,我们可以通过 cuda 启用。如果我们简单地删除相同的内容,就会消除错误。请参阅下面的示例,如果您使用类似的语法模式,则在编译神经网络时删除 Cuda。

1.将代码中指定设备的代码改为cpu

####这里用来表明torch的版本,cpu或者cuda
device = "cpu"

2.将引用cuda的函数修改

from torch import nn
net = nn.Sequential(
    nn.Linear(18*18, 80),
    nn.ReLU(),
    nn.Linear(80, 80),
    nn.ReLU(),
    nn.Linear(80, 10),
    nn.LogSoftmax()
).cuda()

正确的做法是——
在这里插入图片描述
断言错误 torch 未使用启用 cuda 的解决方案进行编译

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值