pytorch gpu版本安装_python 3.8 安装 pytorch gpu

作者：IT小白 | 2024-04-16 13:06:22

踩

python 3.8 安装 pytorch gpu

Anaconda Prompt中创建一个名为pytorch的环境，环境中用到的python版本为3.8

conda create -n pytorch python=3.8

如图所示

输入conda activate pytorch激活到pytorch环境

conda activate pytorch

pytorch

由于我安装的是cuda11.8版本，所以选择的是对应版本

如果安装的cuda版本低的话可以点击

可以根据自己下载的cuda版本进行选择如果没有显卡或者没有下载cuda可以选择 cpu版本的

复制命令到刚才激活的pytorch环境中

等待安装成功

验证pytorch是否安装成功


python
import torch
torch.__version__
torch.cuda.is_available()

输入上述代码，成功如图所示

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/IT小白/article/detail/434349?site