深度学习环境配置cuda12.1 python3.9 pytorch2.2.1安装记录

Bsqfke

于 2024-06-21 21:32:57 发布

阅读量333

点赞数 3

文章标签：深度学习人工智能 python

本文链接：https://blog.csdn.net/Bsqfke/article/details/139870337

版权

cuda12.1 python3.9 pytorch2.2.1安装记录

pytorch官网

https://pytorch.org/get-started/locally/

cuda12.1 python3.9 pytorch2.2.1

出现报错 unsuccessful initial attempt using frozen solve. Retrying with flexible solve.

参考文章：https://zhuanlan.zhihu.com/p/690255462
执行命令：

执行命令

# 创建
conda create -n py39torch221cuda121 python=3.9
# 激活
conda activate py39torch221cuda121
# 下载解码器，解决unsuccessful initial attempt using frozen solve.
# conda install -n py39torch221cuda121 conda-libmamba-solver
conda install conda-libmamba-solver
# 安装库 -c用于指定源位置
conda install pytorch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 pytorch-cuda=12.1 -c pytorch -c nvidia