1.安装Anaconda,配置环变量
在清华源镜像地址下载:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
我的windows选择Anaconda3-2024.10-1-Windows-x86_64.exe这个版本
正常安装,可放置在D盘,
第二个选项不用选
配置环境变量,右击此电脑,属性,进去点击如下图标
将如下三条路径添加到环境变量中(在anaconda的安装路径中)
验证安装
点击windows图标,找到Anaconda Prompt
输入 conda --version
出现conda 24.9.2
即安装成功
2.安装CUDA
选择显卡对应的版本的cuda,下面是所有版本的cuda官网
CUDA Toolkit Archive | NVIDIA Developer
我选择下载cuda的11.8版本,点击Download(3.0G)
CUDA Toolkit 11.8 Downloads | NVIDIA Developer
正常安装前面后,选择自定义安装,仅选择cuda,且VS不用选,安装位置默认在C盘即可。
配置环境变量,同上将如下4条路径添加到环境变量中(在cuda的安装路径中)
验证安装
点击win+R,输入cmd.
输入nvcc -V,出现如下代表安装成功
3.安装pytorch
首先创建一个虚拟环境,打开Anaconda Prompt
输入conda create -n pinn(虚拟环境名) python=3.8
enter后,输入y,进入pinn虚拟空间,
输入conda activate pinn
然后找pytorch下载链接
选择和cuda版本对应的版本
PyTorch(进官网可能需要梯子)
我选择的是以前版本2.3.1
Previous PyTorch Versions | PyTorch
最后输入
conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-cuda=11.8 -c pytorch -c nvidia
enter后,输入y,下载中断,继续复制链接下载,直至成功
验证安装,在pinn虚拟空间中,依次输入
python
import torch
torch.cuda.is_available()
出现True即成功
4.安装pycharm
下载社区版,正常安装,放在D盘,
Download PyCharm: The Python IDE for data science and web development by JetBrains
5.连接pytorch和pycharm
新建项目,放在D盘
配置环境,找到conda.bat文件
等待环境创建完成
添加新的解释器,选择虚拟环境pinn.
等待环境创建完成
验证pytorch和pycharm是否连接成功
新建一个python文件,输入以下内容
import torch
print(torch.cuda.is_available())
右键运行出现True即成功。