自学内容网 自学内容网

windows cuda12.1 pytorch gpu环境配置


安装cuda12.1

nvcc -V

conda创建pythong3.10环境

conda create -n llama3_env python=3.10
conda activate llama3_env

安装pytorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

gpu - Pytorch version for cuda 12.2 - Stack Overflow


原文地址:https://blog.csdn.net/hebian1994/article/details/142435033

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!