文章目录
一、显卡驱动安装
搜索适合你电脑的显卡驱动,安装studio版本。
二、安装anaconda
具体安装步骤参考其他教程。
二、cuda安装
选择你的系统支持的cuda。
三、查看path是否有添加成果
进入 :我的电脑-属性-高级系统设置
进入环境变量。
可以安装多个cuda版本。
查看cuda版本
nvcc -V
四、安装pytorch
更新pip
安装pytorch
进入pytorch官网 https://pytorch.org/
选择适合系统的pyrorch版本。
五、安装pycharm
请查看其他教程。
配置pycharm编译环境和配套的包。
六、查看pytorch 和 gpu是否可以使用
import torch
print(torch.__version__)
print('g:',torch.cuda.is_available())
成功了 !!!