Anaconda虚拟环境安装torch-gpu
经过好几天的尝试,终于成功了=_=
步骤如下所示:
step1·打开cmd,输入nvidia-smi
查看自己电脑支持的CUDA版本,我的版本是12.8,那么安装的cuda版本应当<=12.8
step2·通过conda activate intern
激活虚拟环境(activate后面的是虚拟环境名称),该虚拟环境使用3.11.13版本的python(3.11.X不支持2.5.X的CUDA)
step3·通过conda search cudatoolkit --info
查看支持的cudatoolkit
版本,cudatoolkit
可视为conda版的cuda。发现最新版本是11.8.0,那么conda install cudatoolkit=11.8.0
安装即可。
step4·进入torch官网,根据安装的11.8.0,我们在cmd中选择pip3命令进行安装。同时,由于直接这么安装速度很慢,因此选择使用阿里源实现,不开加速器(2025.6.24 20:00左右,下载速度为14.XMB/s,大概4min下完)
pip3 install torch torchvision torchaudio -f https://mirrors.aliyun.com/pytorch-wheels/cu118
step5·在intern
虚拟环境中,先激活虚拟环境,然后可在一个python文件中(这里我使用vscode打开),通过下列代码测试:
import torch
print(torch.cuda.is_available())
结果若是True,表示成功安装!