一、流程
1、学习Pytorch\Pycharm\Anaconda等概念,理解Anaconda\Pycharm\GPU等在计算机内如何协同工作。
2、判断电脑GPU是否为英伟达,下载Anaconda,创建环境,激活环境。
3、在Anaconda中下载Pytorch并测试。
4、下载Pycharm,新建项目并测试。
二、详细
Pytorch及前导(加粗的源是b站up提供的,大概率好使,其他的是chatgpt和其他地方搜罗来的,大概率不可用)
我的设备:RTX4070SUPER算力8.9(英伟达官网查询,需要vpn)
CUDA Driver Version12.7
所以适用CUDA:11.8,12.0-12.7
确定了是12.4(后面有conda下载时的选配图)
conda install pytorch torchvision torchaudio cudatoolkit=12.4 -c https://mirrors.tuna.tsinghua.edu.cn
https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
清华源:https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda install pytorch=12.4 -c https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/win-64/
torchvision torchaudio cudatoolkit
不知道是不是版本的不同,分享一下现在pycharm的配置方法吧,在点击添加解释器之后得先选上anaconda\Scripts\conda.exe,先别选自己创建的虚拟环境,然后点右边加载环境。这时候下边就有选项,在那里可以直接选到创建的虚拟环境
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
conda install pytorch torchvision