1. 安装Anaconda3
官网下载Anaconda3:https://www.anaconda.com/distribution/
运行下载好的.exe文件
Win+R 调出运行对话框,输入 cmd 回车,输入 python,如果出现python版本信息,表明安装成功。
添加环境变量:高级系统设置 -> 环境变量
2. 查看电脑显卡信息
以 Win10 为例,控制面板 -> NVIDIA控制面板 -> 帮助 -> 系统信息 -> 组件
3. 创建PyTorch环境
卸载原来的pytorch和torchvision,之前没有安装则忽略这一步。
Win+R,输入cmd,pip uninstall torch,结束后再输入pip uninstall torchvision
打开Anaconda Prompt:
创建PyTorch环境
conda create -n pytorch python=3.6
激活PyTorch环境
activate pytorch
退出:deactivate pytorch
4. PyTorch安装
官网下载PyTorch:https://pytorch.org/
p.s. 注意对照版本:)
在Anaconda Prompt中激活PyTorch环境
activate pytorch
安装PyTorch
pip install torch==1.4.0+cu92 torchvision==0.5.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
如果安装太慢,可以现在「 源网站 」中下载好,用pip安装
上述版本的下载链接如下:
https://download.pytorch.org/whl/cu92/torch-1.4.0%2Bcu92-cp36-cp36m-win_amd64.whl
https://download.pytorch.org/whl/cu92/torchvision-0.5.0%2Bcu92-cp36-cp36m-win_amd64.whl
pip install (路径)/torch-1.4.0+cu92-cp36-cp36m-win_amd64.whl
pip install (路径)/torchvision-0.5.0+cu92-cp36-cp36m-win_amd64.whl
5. 验证安装成功
python
>import torch
>import torchvision
均未报错 ,则说明安装成功。
6. PyCharm中配置PyTorch
新建一个PyCharm工程,然后File -> Settings -> Project Interpreter, 这时会发现项目解释器是PyCharm自带的或者是Anaconda自带的python.exe。然而,我们需要使用pytorch的库,所以点击右边代表设置符号,点击add, 选中Existing environment,找到目录并选中:D:\Anaconda\envs\pytorch\python.exe 作为该项目的解释器。
验证Pytorch是否导入Pycharm环节,输入:
from future import print_function
import torch
x = torch.rand(5, 3)
print(x)
点击运行没报错就说明可以了
验证Pytorch是否可以使用GPU和CUDA
import torch
torch.cuda.is_available()
点击运行没报错,输出 True 就说明配置成功了。
总结
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
RTX 5090要首发 性能要翻倍!三星展示GDDR7显存
三星在GTC上展示了专为下一代游戏GPU设计的GDDR7内存。
首次推出的GDDR7内存模块密度为16GB,每个模块容量为2GB。其速度预设为32 Gbps(PAM3),但也可以降至28 Gbps,以提高产量和初始阶段的整体性能和成本效益。
据三星表示,GDDR7内存的能效将提高20%,同时工作电压仅为1.1V,低于标准的1.2V。通过采用更新的封装材料和优化的电路设计,使得在高速运行时的发热量降低,GDDR7的热阻比GDDR6降低了70%。